Searched defs:rhs_shard_size (Results 1 – 1 of 1) sorted by relevance
333 int64 rhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local363 int64 rhs_shard_size = rhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local642 int64 rhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local661 int64 rhs_shard_size = rhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local