Searched refs:ret_index (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | inspecting_placer.cc | 41 for (int ret_index = 0; ret_index < output_groups.size(); ++ret_index) { in DebugString() local
42 int group_id = output_groups[ret_index]; in DebugString()
43 group_members[group_id].push_back(strings::StrCat("o:", ret_index)); in DebugString()
|
| D | process_function_library_runtime.cc | 1050 int ret_index = comp_data.ret_indices[j]; in GetOutputDevices() local
1051 if (data->ret_types_[ret_index] == DT_RESOURCE) { in GetOutputDevices()
1052 (*output_devices)[ret_index] = target_device; in GetOutputDevices()
1054 (*output_devices)[ret_index] = in GetOutputDevices()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 117 for (int ret_index = 0; ret_index < fetches.size(); ++ret_index) { in AddRetvalNodes() local
118 const tf2xla::TensorId& id = fetches[ret_index].id(); in AddRetvalNodes()
132 NodeBuilder(absl::StrCat("_retval_", ret_index), in AddRetvalNodes()
136 .Attr("index", ret_index) in AddRetvalNodes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 234 std::vector<int64> ret_index; in TileIndexForDevice() local
237 ret_index = {index.begin(), index.end()}; in TileIndexForDevice()
240 CHECK(!ret_index.empty()); in TileIndexForDevice()
242 ret_index.pop_back(); in TileIndexForDevice()
244 return ret_index; in TileIndexForDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.cc | 174 for (const int ret_index : function_data->ret_indices.value()) { in Run() local
175 request->add_output_num(ret_index); in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 340 bool ints_on_device, int* ret_index) { in AddReturnNode() argument
374 AddAttr("index", (*ret_index)++, gnode); in AddReturnNode()
797 int ret_index = 0; in InstantiateFunction() local
800 &ret_index); in InstantiateFunction()
1887 std::unordered_map<string, string> ret_index; in Define() local
1889 ret_index[a.name()] = a.name(); in Define()
1904 const auto iter = ret_index.find(a); in Define()
1905 CHECK(iter != ret_index.end()) in Define()
1924 ret_index[src.ret[i]] = in Define()
1933 const auto iter = ret_index.find(r.name()); in Define()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_variable_runtime_reformatting.cc | 256 int64_t ret_index = execute.device_var_updates_indices() in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
260 arg_metadata->set_retval_index_for_sharding(ret_index); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 522 int ret_index = iter->second; in RecordResult() local
530 builder.Attr("index", ret_index); in RecordResult()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1522 int ret_index; in RemoveEdgesBetweenArgAndRetval() local
1523 TF_RETURN_IF_ERROR(GetNodeAttr(e->dst()->def(), "index", &ret_index)); in RemoveEdgesBetweenArgAndRetval()
1535 int output_index = replica_id * old_num_outputs + ret_index; in RemoveEdgesBetweenArgAndRetval()
|