Searched defs:dev_mgr (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_resolver_local.h | 31 explicit DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal()
|
| D | collective_rma_local.h | 28 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal()
|
| D | base_collective_executor.h | 100 const DeviceMgr* dev_mgr, const string* gpu_ring_order, in BaseCollectiveExecutor()
|
| D | collective_executor_mgr.cc | 27 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveExecutorMgr()
|
| D | buf_rendezvous.h | 44 explicit BufRendezvous(uint64 step_id, const DeviceMgr* dev_mgr) in BufRendezvous()
|
| D | collective_util.cc | 32 Status InitializeDeviceAndLocality(const DeviceMgr* dev_mgr, in InitializeDeviceAndLocality()
|
| D | permuter_test.cc | 51 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA()
|
| D | ring_gatherer_test.cc | 47 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA()
|
| D | ring_reducer_test.cc | 48 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA()
|
| D | collective_param_resolver_local.cc | 42 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverLocal()
|
| D | hierarchical_tree_broadcaster_test.cc | 141 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.h | 30 const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessDistributed()
|
| D | device_resolver_distributed.cc | 23 DeviceResolverDistributed::DeviceResolverDistributed(const DeviceMgr* dev_mgr) { in DeviceResolverDistributed()
|
| D | rpc_collective_executor_mgr.cc | 29 const ConfigProto& config, const DeviceMgr* dev_mgr, in RpcCollectiveExecutorMgr()
|
| D | collective_rma_distributed_test.cc | 65 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker()
248 DeviceMgr* dev_mgr = new StaticDeviceMgr(std::move(devices)); in DefineWorker() local
|
| D | collective_param_resolver_distributed.cc | 92 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverDistributed()
|
| D | collective_param_resolver_distributed_test.cc | 52 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 171 const DeviceMgr* dev_mgr, OpKernelContext* ctx, in CollectiveContext()
|
| D | collective.h | 367 const DeviceMgr* dev_mgr; // Not owned member
|