Home
last modified time | relevance | path

Searched defs:ret_index (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils_test.cc94 void SubGraph(Graph* subgraph, DataType dtype, int arg_index, int ret_index) { in SubGraph()
Dprocess_function_library_runtime.cc722 int ret_index = comp_data.ret_indices_[j]; in GetOutputDevices() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc125 for (int ret_index = 0; ret_index < fetches.size(); ++ret_index) { in AddRetvalNodes() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc144 std::vector<int64> ret_index; in TileIndexForDevice() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc314 bool ints_on_device, int* ret_index) { in AddReturnNode()
750 int ret_index = 0; in InstantiateFunction() local
1612 std::unordered_map<string, string> ret_index; in Define() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc815 int ret_index = iter->second; in RecordResult() local