Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.h38 PaddingConfig MakeSymmetricPadding(absl::Span<const int64> sizes);
Dwindow_util.cc55 PaddingConfig MakeSymmetricPadding(absl::Span<const int64> sizes) { in MakeSymmetricPadding() function
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc578 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PadFromPartialReplicateShape()
611 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PadFromPartialReplicateShape()
Dconvolution_handler.cc351 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
Ddot_handler.cc805 window_util::MakeSymmetricPadding(min_padding); in PartitionBaseCase()
816 window_util::MakeSymmetricPadding(min_padding); in PartitionBaseCase()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc5089 PaddingConfig padding = window_util::MakeSymmetricPadding( in TEST_P()