Searched refs:UniqueDevice (Results 1 – 8 of 8) sorted by relevance
123 auto unique_device = tuple_element_sharding.UniqueDevice(); in UsedDevices()130 auto unique_device = UniqueDevice(); in UsedDevices()244 absl::optional<int64> HloSharding::UniqueDevice() const { in UniqueDevice() function in xla::HloSharding251 auto device = tuple_sharding.UniqueDevice(); in UniqueDevice()266 auto device = UniqueDevice(); in GetUniqueDevice()
155 absl::optional<int64> UniqueDevice() const;161 bool HasUniqueDevice() const { return UniqueDevice().has_value(); } in HasUniqueDevice()
196 return lhs->UniqueDevice() != rhs.UniqueDevice() in AssignLeafSharding()
1154 return sharding_->UniqueDevice(); in sharding_unique_device()
869 auto device = sharding.UniqueDevice(); in SetupCopiedInstruction()
128 TEST(PickDeviceForXla, UniqueDevice) { in TEST() argument
15065 using UniqueDevice = UniqueHandle<Device, DispatchLoaderStatic>; typedef
45095 using UniqueDevice = UniqueHandle<Device,DispatchLoaderStatic>; typedef