Home
last modified time | relevance | path

Searched defs:retval_index (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dsubgraph.h144 const DeviceAttributes* device_info, int32_t retval_index) in RetvalFetchRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_sharding_identification_pass.cc264 int retval_index = v.getInt(); in ExtractAliases() local
276 int retval_index = value.cast<OpResult>().getResultNumber(); in GetXlaShardingFromAlias() local
Dresource_op_lifting.cc1080 for (int64_t retval_index : retval_indices_to_preserve) in HandlePartitionedCallOpCallee() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3722 int retval_index = core_retval_nums[core][i]; in BuildExecuteNodes() local