Searched refs:HasUniqueDevice (Results 1 – 4 of 4) sorted by relevance
205 bool HasUniqueDevice() const { return UniqueDevice().has_value(); } in HasUniqueDevice() function
68 EXPECT_FALSE(sharding.HasUniqueDevice()); in TEST_F()149 EXPECT_FALSE(sharding.HasUniqueDevice()); in TEST_F()
1640 bool is_spatially_partitioned = !sharding->HasUniqueDevice(); in NormalizeDomain()1644 [](const HloSharding& s) { return !s.HasUniqueDevice(); }); in NormalizeDomain()
1058 CHECK(sharding.HasUniqueDevice()); in Broadcast()1387 HloSharding sharding = hlo->sharding().HasUniqueDevice() in DefaultAction()2179 TF_RET_CHECK(hlo->sharding().HasUniqueDevice()); in HandleSingleDevice()2516 if (hlo->sharding().HasUniqueDevice()) { in HandleInfeed()2899 if (hlo->sharding().HasUniqueDevice()) { in HandleOutfeed()3044 if (hlo->sharding().HasUniqueDevice()) { in HandleRng()