Searched defs:source_sharding (Results 1 – 2 of 2) sorted by relevance
1670 const Shape& target_shape, const HloSharding& source_sharding, in CreateMatchingShardingOnDims()
3235 const auto source_sharding = in HandleGetTupleElement() local