Searched defs:shard_size_with_halo (Results 1 – 2 of 2) sorted by relevance
468 int64 shard_size_with_halo = new_window.dimensions(i).size(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local694 int64 shard_size_with_halo = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
927 int64 shard_size_with_halo, int64 dim, const HloSharding& target, in ExchangeHaloAndGetValidData()