Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dinspecting_placer.cc41 for (int ret_index = 0; ret_index < output_groups.size(); ++ret_index) { in DebugString() local
Dprocess_function_library_runtime.cc1050 int ret_index = comp_data.ret_indices[j]; in GetOutputDevices() local
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.cc174 for (const int ret_index : function_data->ret_indices.value()) { in Run() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc117 for (int ret_index = 0; ret_index < fetches.size(); ++ret_index) { in AddRetvalNodes() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_variable_runtime_reformatting.cc256 int64_t ret_index = execute.device_var_updates_indices() in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc340 bool ints_on_device, int* ret_index) { in AddReturnNode()
797 int ret_index = 0; in InstantiateFunction() local
1887 std::unordered_map<string, string> ret_index; in Define() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc234 std::vector<int64> ret_index; in TileIndexForDevice() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc522 int ret_index = iter->second; in RecordResult() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1522 int ret_index; in RemoveEdgesBetweenArgAndRetval() local