Searched defs:dev_mgr (Results 1 – 15 of 15) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | device_resolver_local.h | 31 explicit DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal()
|
| H A D | collective_rma_local.h | 28 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal()
|
| H A D | base_collective_executor.h | 100 const DeviceMgr* dev_mgr, in BaseCollectiveExecutor()
|
| H A D | buf_rendezvous.h | 44 explicit BufRendezvous(uint64 step_id, const DeviceMgr* dev_mgr) in BufRendezvous()
|
| H A D | collective_util.cc | 32 Status InitializeDeviceAndLocality(const DeviceMgr* dev_mgr, in InitializeDeviceAndLocality()
|
| H A D | collective_executor_mgr.cc | 30 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveExecutorMgr()
|
| H A D | collective_test_util.cc | 33 FailTestRMA::FailTestRMA(const DeviceMgr* dev_mgr, in FailTestRMA()
|
| H A D | collective_param_resolver_local.cc | 46 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverLocal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_rma_distributed.h | 30 const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessDistributed()
|
| H A D | device_resolver_distributed.cc | 23 DeviceResolverDistributed::DeviceResolverDistributed(const DeviceMgr* dev_mgr) { in DeviceResolverDistributed()
|
| H A D | collective_rma_distributed_test.cc | 81 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker()
297 DeviceMgr* dev_mgr = new StaticDeviceMgr(std::move(devices)); in DefineWorker() local
|
| H A D | rpc_collective_executor_mgr.cc | 29 const ConfigProto& config, const DeviceMgr* dev_mgr, in RpcCollectiveExecutorMgr()
|
| H A D | collective_param_resolver_distributed.cc | 90 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverDistributed()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | collective.cc | 157 const DeviceMgr* dev_mgr, OpKernelContext* ctx, in CollectiveContext()
|
| H A D | collective.h | 394 const DeviceMgr* dev_mgr; // Not owned member
|