Searched refs:lhs_shard_size (Results 1 – 1 of 1) sorted by relevance
331 int64 lhs_shard_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local336 lhs_shard_sizes[i] = lhs_shard_size; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()362 int64 lhs_shard_size = lhs_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()424 new_padding_low - lhs_shard_size; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()435 lhs_shard_size + 2 * new_padding_low_max); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()442 lhs_shard_size + new_padding_low_max); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()[all …]