Searched refs:ret_index (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 125 for (int ret_index = 0; ret_index < fetches.size(); ++ret_index) { in AddRetvalNodes() local
126 const tf2xla::TensorId& id = fetches[ret_index].id(); in AddRetvalNodes()
140 NodeBuilder(absl::StrCat("_retval_", ret_index), kRetvalOp) in AddRetvalNodes()
143 .Attr("index", ret_index) in AddRetvalNodes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 144 std::vector<int64> ret_index; in TileIndexForDevice() local
147 ret_index = {index.begin(), index.end()}; in TileIndexForDevice()
150 CHECK(!ret_index.empty()); in TileIndexForDevice()
151 return ret_index; in TileIndexForDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 314 bool ints_on_device, int* ret_index) { in AddReturnNode() argument
347 AddAttr("index", (*ret_index)++, gnode); in AddReturnNode()
750 int ret_index = 0; in InstantiateFunction() local
753 &ret_index); in InstantiateFunction()
1612 std::unordered_map<string, string> ret_index; in Define() local
1614 ret_index[a.name()] = a.name(); in Define()
1629 const auto iter = ret_index.find(a); in Define()
1630 CHECK(iter != ret_index.end()) in Define()
1649 ret_index[src.ret[i]] = in Define()
1658 const auto iter = ret_index.find(r.name()); in Define()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils_test.cc | 94 void SubGraph(Graph* subgraph, DataType dtype, int arg_index, int ret_index) { in SubGraph() argument
99 auto dx_retval = ops::_Retval(s1.WithOpName("retval1"), id_x, ret_index); in SubGraph()
|
| D | process_function_library_runtime.cc | 722 int ret_index = comp_data.ret_indices_[j]; in GetOutputDevices() local
724 (*output_devices)[ret_index] = target_device; in GetOutputDevices()
726 (*output_devices)[ret_index] = in GetOutputDevices()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 815 int ret_index = iter->second; in RecordResult() local
823 builder.Attr("index", ret_index); in RecordResult()
|