Searched defs:new_sharding (Results 1 – 2 of 2) sorted by relevance
1481 HloSharding new_sharding = in GetShardingFromUser() local1493 HloSharding new_sharding = in GetShardingFromUser() local1773 HloSharding new_sharding = operand->sharding().GetSubSharding( in InferShardingFromOperands() local1854 HloSharding new_sharding = HloSharding::Tuple(shape, sub_shardings); in InferShardingFromOperands() local1901 HloSharding new_sharding = in InferShardingFromOperands() local1943 HloSharding new_sharding = in InferShardingFromOperands() local2047 std::optional<HloSharding> new_sharding = in InferShardingFromOperands() local2140 HloSharding new_sharding = hlo_sharding_util::GatherOutputSharding( in InferShardingFromOperands() local
1857 auto new_sharding = sharding.ReplicateOnLastTileDim() in SplitShardingDimension() local