Home
last modified time | relevance | path

Searched defs:resource_arg_index (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dresource_spmd_expander.cc59 const int resource_arg_index, in ValidateAndAssignResourceInputLayout()
64 std::string layout) { in ValidateAndAssignResourceInputLayout()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dresource_op_lifting.cc946 int64_t resource_arg_index = entry.getFirst(); in HandleCaseOrIfOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc2036 int resource_arg_index = node->attrs().Find("index")->i(); in InferShapesWithResourceVar() local