Searched defs:GetDeviceSet (Results 1 – 4 of 4) 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