Searched refs:arg_cores (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 92 std::map<int, int> arg_cores; in ComputeArgAndRetvalCores() local
101 arg_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
260 auto it = arg_cores.find(resource->arg_num()); in BuildComputation()
261 const int core = it == arg_cores.end() ? -1 : it->second; in BuildComputation()
705 const std::map<int, int>& arg_cores, in BuildArguments() argument
795 auto it = arg_cores.find(parameter); in BuildArguments()
796 const int core = it == arg_cores.end() ? 0 : it->second; in BuildArguments()
819 auto it = arg_cores.find(i); in BuildArguments()
[all …]
|
| D | xla_compiler.h | 434 const std::map<int, int>& arg_cores,
|