| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | rpc_collective_executor_mgr.cc | 30 std::unique_ptr<DeviceResolverDistributed> dev_resolver, in RpcCollectiveExecutorMgr() argument
34 : CollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in RpcCollectiveExecutorMgr()
176 auto dev_resolver = std::make_unique<DeviceResolverDistributed>(device_mgr); in CreateProdRpcCollectiveExecutorMgr() local
178 config, device_mgr, dev_resolver.get(), nccl_communicator.get(), in CreateProdRpcCollectiveExecutorMgr()
181 config, device_mgr, std::move(dev_resolver), std::move(param_resolver), in CreateProdRpcCollectiveExecutorMgr()
|
| H A D | collective_rma_distributed.h | 30 const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessDistributed() argument
33 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
|
| H A D | collective_param_resolver_distributed.cc | 91 DeviceResolverDistributed* dev_resolver, in CollectiveParamResolverDistributed() argument
94 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, in CollectiveParamResolverDistributed()
|
| H A D | collective_param_resolver_distributed.h | 33 DeviceResolverDistributed* dev_resolver,
|
| H A D | rpc_collective_executor_mgr.h | 39 std::unique_ptr<DeviceResolverDistributed> dev_resolver,
|
| H A D | collective_rma_distributed_test.cc | 281 DeviceResolverDistributed* dev_resolver = dev_resolver_item.second; in ResolveDeviceAttributes() local
283 TF_CHECK_OK(dev_resolver->UpdateDeviceAttributes(item.second)); in ResolveDeviceAttributes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | collective_rma_local.h | 29 DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessLocal() argument
32 dev_resolver_(dev_resolver), in CollectiveRemoteAccessLocal()
|
| H A D | collective_executor_mgr.cc | 31 std::unique_ptr<DeviceResolverInterface> dev_resolver, in CollectiveExecutorMgr() argument
35 dev_resolver_(std::move(dev_resolver)), in CollectiveExecutorMgr()
|
| H A D | collective_test_util.cc | 34 DeviceResolverInterface* dev_resolver, int64_t step_id) in FailTestRMA() argument
35 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| H A D | collective_executor_mgr.h | 30 std::unique_ptr<DeviceResolverInterface> dev_resolver,
|
| H A D | collective_test_util.h | 40 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver,
|
| H A D | collective_param_resolver_local.h | 45 DeviceResolverInterface* dev_resolver,
|
| H A D | collective_param_resolver_local.cc | 47 DeviceResolverInterface* dev_resolver, in CollectiveParamResolverLocal() argument
51 dev_resolver_(dev_resolver), in CollectiveParamResolverLocal()
|