Home
last modified time | relevance | path

Searched refs:new_padding_low_max (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc419 int64 new_padding_low_max = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
421 int64 new_padding_low = new_padding_low_max; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
431 ->set_edge_padding_low(new_padding_low_max); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
433 ->set_edge_padding_high(new_padding_low_max); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
435 lhs_shard_size + 2 * new_padding_low_max); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
438 MultiplyAddDivideOffsetCalculation(0, new_padding_low_max, 1)) - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
442 lhs_shard_size + new_padding_low_max); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()