Searched refs:device_resolver (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | collective_executor_mgr.cc | 100 auto device_resolver = std::make_unique<DeviceResolverLocal>(device_mgr); in CreateProdLocalCollectiveExecutorMgr() local
102 config, device_mgr, device_resolver.get(), nccl_communicator.get(), in CreateProdLocalCollectiveExecutorMgr()
105 config, device_mgr, std::move(device_resolver), std::move(param_resolver), in CreateProdLocalCollectiveExecutorMgr()
|
| H A D | collective_test_util.cc | 170 test_env->device_resolver = in CreateCollectiveTestEnv()
176 test_env->device_mgr.get(), test_env->device_resolver.get(), kStepId); in CreateCollectiveTestEnv()
|
| H A D | collective_test_util.h | 82 std::unique_ptr<DeviceResolverInterface> device_resolver; member
|