Searched refs:GetDeviceSet (Results 1 – 6 of 6) sorted by relevance
51 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet() function
104 virtual const DeviceSet* GetDeviceSet() const { return nullptr; } in GetDeviceSet() function
46 const DeviceSet* GetDeviceSet() const override { return device_set_.get(); } in GetDeviceSet() function
133 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet() function in tensorflow::tensorrt::convert::FakeCluster
173 for (const auto dev : cluster->GetDeviceSet()->devices()) { in PrintDebugInfo()
734 if (params.cluster == nullptr || params.cluster->GetDeviceSet() == nullptr || in GetDeviceAndAllocator()760 auto device_set = params.cluster->GetDeviceSet(); in GetDeviceAndAllocator()