Searched defs:rhs_shard_size (Results 1 – 1 of 1) sorted by relevance
331 int64_t rhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local361 int64_t rhs_shard_size = rhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local641 int64_t rhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local660 int64_t rhs_shard_size = rhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local