Searched refs:GetDeviceSet (Results 1 – 7 of 7) sorted by relevance
41 static DeviceSet *GetDeviceSet() in GetDeviceSet() function76 const DeviceSet *deviceSet = GetDeviceSet(); in IsValidDevice()83 ASSERT(GetDeviceSet()->find(this) == GetDeviceSet()->end()); in Device()84 GetDeviceSet()->insert(this); in Device()90 ASSERT(GetDeviceSet()->find(this) != GetDeviceSet()->end()); in ~Device()91 GetDeviceSet()->erase(this); in ~Device()
53 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
59 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet() function in tensorflow::tensorrt::convert::FakeCluster
171 for (const auto dev : cluster->GetDeviceSet()->devices()) { in PrintDebugInfo()
558 if (params.cluster == nullptr || params.cluster->GetDeviceSet() == nullptr || in GetDeviceAndAllocator()577 auto device_set = params.cluster->GetDeviceSet(); in GetDeviceAndAllocator()