Searched defs:sub_sharding (Results 1 – 3 of 3) sorted by relevance
532 for (const auto& sub_sharding : sharding_new_metadata.tuple_elements()) { in TEST_F() local567 for (const auto& sub_sharding : sharding_no_metadata.tuple_elements()) { in TEST_F() local
946 for (const HloSharding& sub_sharding : tuple_sharding.tuple_elements()) { in TEST_P() local2008 for (const HloSharding& sub_sharding : instr->sharding().tuple_elements()) { in TEST_P() local2894 for (const HloSharding& sub_sharding : in TEST_P() local3046 for (const HloSharding& sub_sharding : in TEST_P() local
1424 auto sub_sharding = reduce_window->sharding().GetSubSharding( in GetShardingFromUser() local1461 auto sub_sharding = user.sharding().GetSubSharding( in GetShardingFromUser() local