Searched refs:retval_cores (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 93 std::map<int, int> retval_cores; in ComputeArgAndRetvalCores() local
108 TF_ASSIGN_OR_RETURN(retval_cores[index], get_sharding_for_node(n)); in ComputeArgAndRetvalCores()
109 retval_cores[index] = core; in ComputeArgAndRetvalCores()
112 return std::make_pair(std::move(arg_cores), std::move(retval_cores)); in ComputeArgAndRetvalCores()
160 const std::map<int, int>& arg_cores, const std::map<int, int>& retval_cores, in BuildComputation() argument
209 auto it = retval_cores.find(i); in BuildComputation()
211 builder, it == retval_cores.end() in BuildComputation()
221 } else if (it != retval_cores.end()) { in BuildComputation()
1085 std::map<int, int> retval_cores; in CompileGraph() local
1086 TF_ASSIGN_OR_RETURN(std::tie(arg_cores, retval_cores), in CompileGraph()
[all …]
|