Searched refs:ret_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 123 for (int ret_index = 0; ret_index < fetches.size(); ++ret_index) { in AddRetvalNodes() local
124 const tf2xla::TensorId& id = fetches[ret_index].id(); in AddRetvalNodes()
138 NodeBuilder(strings::StrCat("_retval_", ret_index), kRetvalOp) in AddRetvalNodes()
141 .Attr("index", ret_index) in AddRetvalNodes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 80 std::vector<int64> ret_index; in TileIndexForDevice() local
83 ret_index = {index.begin(), index.end()}; in TileIndexForDevice()
86 CHECK(!ret_index.empty()); in TileIndexForDevice()
87 return ret_index; in TileIndexForDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 305 int* ret_index) { in AddReturnNode() argument
334 AddAttr("index", (*ret_index)++, gnode); in AddReturnNode()
689 int ret_index = 0; in InstantiateFunction() local
691 s = helper.AddReturnNode(ret_def, attr_values, fdef.ret(), &ret_index); in InstantiateFunction()
1259 std::unordered_map<string, string> ret_index; in Define() local
1261 ret_index[a.name()] = a.name(); in Define()
1276 const auto iter = ret_index.find(a); in Define()
1277 CHECK(iter != ret_index.end()) in Define()
1296 ret_index[src.ret[i]] = in Define()
1304 const auto iter = ret_index.find(r.name()); in Define()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 696 int ret_index = iter->second; in RecordResult() local
703 builder.Attr("index", ret_index); in RecordResult()
|