Searched refs:UniqueDevice (Results 1 – 8 of 8) sorted by relevance
212 auto unique_device = tuple_element_sharding.UniqueDevice(); in UsedDevices()219 auto unique_device = UniqueDevice(); in UsedDevices()352 absl::optional<int64> HloSharding::UniqueDevice() const { in UniqueDevice() function in xla::HloSharding359 auto device = tuple_sharding.UniqueDevice(); in UniqueDevice()374 auto device = UniqueDevice(); in GetUniqueDevice()
199 absl::optional<int64> UniqueDevice() const;205 bool HasUniqueDevice() const { return UniqueDevice().has_value(); } in HasUniqueDevice()
196 return lhs->UniqueDevice() != rhs.UniqueDevice() in AssignLeafSharding()
1397 return sharding_->UniqueDevice(); in sharding_unique_device()
922 auto device = sharding.UniqueDevice(); in SetupCopiedInstruction()
72 TEST(PickDeviceForXla, UniqueDevice) { in TEST() argument
3785 root_sharding.UniqueDevice().has_value()) in PreprocessSharding()3791 param->sharding().UniqueDevice().has_value()) in PreprocessSharding()
183 if (partition_id == *sharding.UniqueDevice()) { in MakeNonPaddedShapeForGivenPartition()