Searched refs:GetUniqueDevice (Results 1 – 3 of 3) sorted by relevance
158 int64 GetUniqueDevice() const;
62 EXPECT_EQ(5, sharding.GetUniqueDevice()); in TEST_F()
265 int64 HloSharding::GetUniqueDevice() const { in GetUniqueDevice() function in xla::HloSharding