Searched refs:GetUniqueDevice (Results 1 – 4 of 4) sorted by relevance
202 int64 GetUniqueDevice() const;
77 EXPECT_EQ(5, sharding.GetUniqueDevice()); in TEST_F()
373 int64 HloSharding::GetUniqueDevice() const { in GetUniqueDevice() function in xla::HloSharding
1062 LiteralUtil::CreateR0<uint32>(sharding.GetUniqueDevice()))); in Broadcast()2180 int64 device = hlo->sharding().GetUniqueDevice(); in HandleSingleDevice()