Searched defs:sub_sharding (Results 1 – 3 of 3) sorted by relevance
525 for (const auto& sub_sharding : sharding_new_metadata.tuple_elements()) { in TEST_F() local560 for (const auto& sub_sharding : sharding_no_metadata.tuple_elements()) { in TEST_F() local
831 for (const HloSharding& sub_sharding : tuple_sharding.tuple_elements()) { in TEST_P() local1800 for (const HloSharding& sub_sharding : instr->sharding().tuple_elements()) { in TEST_P() local2495 for (const HloSharding& sub_sharding : in TEST_P() local2647 for (const HloSharding& sub_sharding : in TEST_P() local
1238 auto sub_sharding = user.sharding().GetSubSharding( in GetShardingFromUser() local1272 auto sub_sharding = user.sharding().GetSubSharding( in GetShardingFromUser() local