Home
last modified time | relevance | path

Searched refs:retval_index (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dsubgraph.h144 const DeviceAttributes* device_info, int32 retval_index) in RetvalFetchRewrite() argument
145 : PruneRewrite(endpoint_name, device_info), retval_index_(retval_index) {} in RetvalFetchRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_op_lifting.cc1147 for (int64_t retval_index : retval_indices_to_preserve) in HandlePartitionedCallOpCallee() local
1148 old_and_new_retvals.push_back(old_return->getOperand(retval_index)); in HandlePartitionedCallOpCallee()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3285 int retval_index = core_retval_nums[core][i]; in BuildExecuteNodes() local
3286 retval_index_to_output_index_mapping[retval_index][core] = i; in BuildExecuteNodes()
3335 retval_index_to_output_index_mapping[retval_index][core_id]; in BuildExecuteNodes()