Home
last modified time | relevance | path

Searched defs:dev_mgr (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_resolver_local.h29 DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal()
Dcollective_rma_local.h27 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal()
Dcollective_executor_mgr.cc27 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveExecutorMgr()
Dcollective_util.cc32 Status InitializeDeviceAndLocality(const DeviceMgr* dev_mgr, in InitializeDeviceAndLocality()
Dbase_collective_executor.h98 int64 step_id, const DeviceMgr* dev_mgr, in BaseCollectiveExecutor()
Dcollective_param_resolver_local.cc41 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverLocal()
Dring_gatherer_test.cc46 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA()
Dring_reducer_test.cc46 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA()
Dhierarchical_tree_broadcaster_test.cc138 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.h26 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed()
Ddevice_resolver_distributed_test.cc35 TestableDeviceResolverDistributed(const DeviceMgr* dev_mgr, in TestableDeviceResolverDistributed()
64 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker()
160 DeviceMgr* dev_mgr = new DeviceMgr(std::move(devices)); in DefineWorker() local
Ddevice_resolver_distributed.cc22 const DeviceMgr* dev_mgr, WorkerCacheInterface* worker_cache, in DeviceResolverDistributed()
Drpc_collective_executor_mgr.cc29 const ConfigProto& config, const DeviceMgr* dev_mgr, in RpcCollectiveExecutorMgr()
Dcollective_param_resolver_distributed_test.cc49 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker()
165 DeviceMgr* dev_mgr = new DeviceMgr(std::move(devices)); in DefineWorker() local
Dcollective_rma_distributed_test.cc63 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker()
220 DeviceMgr* dev_mgr = new DeviceMgr(std::move(devices)); in DefineWorker() local
Dcollective_param_resolver_distributed.cc82 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverDistributed()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_collective_executor_mgr.h36 const ConfigProto& config, const DeviceMgr* dev_mgr, in GdrCollectiveExecutorMgr()
Dgdr_collective_executor_mgr.cc67 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc157 const DeviceMgr* dev_mgr, in CollectiveContext()
Dcollective.h349 const DeviceMgr* dev_mgr; // Not owned variable