Searched refs:CollInstanceParams (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 79 struct CollInstanceParams { struct
100 CollInstanceParams& operator=(const struct CollInstanceParams& other); argument
113 CollInstanceParams instance;
136 virtual void GetDeviceLocalitiesAsync(const CollInstanceParams& inst_params,
|
| D | collective.cc | 57 CollInstanceParams& CollInstanceParams::operator=( in operator =()
58 const CollInstanceParams& other) { in operator =()
87 string CollInstanceParams::ToString() const { in ToString()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed.h | 37 void GetDeviceLocalitiesAsync(const CollInstanceParams& inst_params,
55 void GetDeviceLocalitiesRecursive(const CollInstanceParams& inst_params,
|
| D | device_resolver_distributed.cc | 68 const CollInstanceParams& inst_params, in GetDeviceLocalitiesAsync()
75 const CollInstanceParams& inst_params, in GetDeviceLocalitiesRecursive()
|
| D | collective_param_resolver_distributed.cc | 49 const CollInstanceParams& instance, in CompleteInstanceCall()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_resolver_local.h | 33 void GetDeviceLocalitiesAsync(const CollInstanceParams& ci_params,
|
| D | device_resolver_local.cc | 22 const CollInstanceParams& ci_params, in GetDeviceLocalitiesAsync()
|
| D | collective_rma_local.h | 55 void GetDeviceLocalitiesAsync(const CollInstanceParams& ci_params, in GetDeviceLocalitiesAsync()
|
| D | collective_param_resolver_local.cc | 159 GlobalDeviceMap BuildDevRecs(const CollInstanceParams& ip, in BuildDevRecs()
|