Searched refs:rhs_shard_size (Results 1 – 1 of 1) sorted by relevance
333 int64 rhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local337 rhs_shard_sizes[i] = rhs_shard_size; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()363 int64 rhs_shard_size = rhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local381 rhs_shard_size * wd.window_dilation() - lhs_shard_size, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()387 lhs_shard_size - rhs_shard_size * wd.window_dilation(), in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()388 lhs_shard_size - rhs_shard_size * wd.window_dilation() + in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()416 rhs_shard_size * wd.window_dilation() - lhs_shard_size, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()642 int64 rhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local646 rhs_shard_sizes[i] = rhs_shard_size; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()661 int64 rhs_shard_size = rhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local[all …]