Searched refs:DeviceResolverLocal (Results 1 – 11 of 11) sorted by relevance
27 class DeviceResolverLocal : public DeviceResolverInterface {29 DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal() function31 virtual ~DeviceResolverLocal() {} in ~DeviceResolverLocal()
21 void DeviceResolverLocal::GetDeviceLocalitiesAsync( in GetDeviceLocalitiesAsync()37 void DeviceResolverLocal::GetLocalityAsync(const string& device, in GetLocalityAsync()
41 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in DeviceResolverLocalTest()45 std::unique_ptr<DeviceResolverLocal> drl_;
48 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in CollectiveRemoteAccessLocalTest()56 std::unique_ptr<DeviceResolverLocal> drl_;
43 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in CollectiveParamResolverLocalTest()78 std::unique_ptr<DeviceResolverLocal> drl_;
45 new DeviceResolverLocal(device_mgr_.get())); in CollectiveExecutorMgrTest()
170 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()519 std::unique_ptr<DeviceResolverLocal> dev_resolver_;
192 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()546 std::unique_ptr<DeviceResolverLocal> dev_resolver_;
250 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()715 std::unique_ptr<DeviceResolverLocal> dev_resolver_;
502 new DeviceResolverLocal(device_mgr_.get())); in RunInternal()
85 new DeviceResolverLocal(local_device_mgr())); in EagerContext()