Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc53 PaddingConfig pad_config = MakeNoPaddingConfig(shape.rank()); in PadInstruction() local
61 pad_config.mutable_dimensions(dim)->set_edge_padding_high( in PadInstruction()
70 HloInstruction::CreatePad(new_shape, instr, zero, pad_config)); in PadInstruction()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc578 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PadFromPartialReplicateShape() local
579 pad_config.mutable_dimensions(dim)->set_edge_padding_low(0); in PadFromPartialReplicateShape()
582 pad_config.mutable_dimensions(dim)->set_edge_padding_high(std::max( in PadFromPartialReplicateShape()
586 concat->shape(), zero->shape(), pad_config) in PadFromPartialReplicateShape()
589 padded_concat_shape, concat, zero, pad_config)); in PadFromPartialReplicateShape()
611 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PadFromPartialReplicateShape() local
615 pad_config.mutable_dimensions(dim)->set_edge_padding_low(0); in PadFromPartialReplicateShape()
616 pad_config.mutable_dimensions(dim)->set_edge_padding_high( in PadFromPartialReplicateShape()
623 HloInstruction::CreatePad(padded_shape, result, zero, pad_config)); in PadFromPartialReplicateShape()
Dconvolution_handler.cc351 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
430 pad_config.mutable_dimensions(lhs_dimension) in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
432 pad_config.mutable_dimensions(lhs_dimension) in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
456 HloInstruction::CreatePad(pad_shape, lhs.hlo(), zero, pad_config)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc3424 PaddingConfig pad_config; in HandlePad() local
3437 auto dimension = pad_config.add_dimensions(); in HandlePad()
3445 *pad2->mutable_padding_config() = pad_config; in HandlePad()
4859 const PaddingConfig& pad_config = pad->padding_config(); in HandleReduceWindow() local
4860 if (HasInteriorPadding(pad_config) && window_util::HasBaseDilation(window)) { in HandleReduceWindow()
4902 if (!window_util::HasSymmetricPadding(pad_config)) { in HandleReduceWindow()
4906 if (HasInteriorPadding(pad_config)) { in HandleReduceWindow()
4910 for (int64 i = 0; i < pad_config.dimensions_size(); ++i) { in HandleReduceWindow()
4911 const auto& pad_dimension = pad_config.dimensions(i); in HandleReduceWindow()
4922 const auto& pad_dimension = pad_config.dimensions(i); in HandleReduceWindow()
[all …]
Dhlo_evaluator_typed_visitor.h1565 const PaddingConfig& pad_config = pad->padding_config();
1573 pad_config.dimensions(i).edge_padding_low() +
1574 input_index[i] * (pad_config.dimensions(i).interior_padding() + 1);