Searched refs:inst_params (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed.cc | 68 const CollInstanceParams& inst_params, in GetDeviceLocalitiesAsync() argument
71 GetDeviceLocalitiesRecursive(inst_params, localities, done); in GetDeviceLocalitiesAsync()
75 const CollInstanceParams& inst_params, in GetDeviceLocalitiesRecursive() argument
78 if (i < inst_params.device_names.size()) { in GetDeviceLocalitiesRecursive()
80 GetLocalityAsync(inst_params.device_names[i], inst_params.task_names[i], in GetDeviceLocalitiesRecursive()
82 [this, &inst_params, localities, done](const Status& s) { in GetDeviceLocalitiesRecursive()
87 GetDeviceLocalitiesRecursive(inst_params, localities, in GetDeviceLocalitiesRecursive()
|
| D | device_resolver_distributed.h | 37 void GetDeviceLocalitiesAsync(const CollInstanceParams& inst_params,
55 void GetDeviceLocalitiesRecursive(const CollInstanceParams& inst_params,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 136 virtual void GetDeviceLocalitiesAsync(const CollInstanceParams& inst_params,
|