Home
last modified time | relevance | path

Searched defs:dev_resolver (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Drpc_collective_executor_mgr.cc30 std::unique_ptr<DeviceResolverDistributed> dev_resolver, in RpcCollectiveExecutorMgr()
176 auto dev_resolver = std::make_unique<DeviceResolverDistributed>(device_mgr); in CreateProdRpcCollectiveExecutorMgr() local
H A Dcollective_rma_distributed.h30 const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessDistributed()
H A Dcollective_param_resolver_distributed.cc91 DeviceResolverDistributed* dev_resolver, in CollectiveParamResolverDistributed()
H A Dcollective_rma_distributed_test.cc281 DeviceResolverDistributed* dev_resolver = dev_resolver_item.second; in ResolveDeviceAttributes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_rma_local.h29 DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessLocal()
H A Dcollective_executor_mgr.cc31 std::unique_ptr<DeviceResolverInterface> dev_resolver, in CollectiveExecutorMgr()
H A Dcollective_test_util.cc34 DeviceResolverInterface* dev_resolver, int64_t step_id) in FailTestRMA()
H A Dcollective_param_resolver_local.cc47 DeviceResolverInterface* dev_resolver, in CollectiveParamResolverLocal()