Searched defs:lhs_shard_size (Results 1 – 1 of 1) sorted by relevance
331 int64 lhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local362 int64 lhs_shard_size = lhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local640 int64 lhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local660 int64 lhs_shard_size = lhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local