Searched refs:resource_vars (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.h | 159 const std::map<int, const Tensor*>& resource_vars,
179 const std::map<int, const Tensor*>& resource_vars);
|
| D | xla_launch_util.cc | 223 const std::map<int, const Tensor*>& resource_vars, in PopulateInputs() argument
239 bool is_resource_variable = resource_vars.count(arg_num); in PopulateInputs()
248 ? resource_vars.at(arg_num) in PopulateInputs()
426 const std::map<int, const Tensor*>& resource_vars) { in PopulateOutputs() argument
520 compilation_result->input_mapping, resource_vars, in PopulateOutputs()
567 compilation_result->input_mapping, resource_vars, write.type, in PopulateOutputs()
|