Home
last modified time | relevance | path

Searched defs:padding_high (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_padding_legalization.cc230 int64_t padding_high = backward_conv->window().dimensions(i).padding_high(); in CanonicalizeBackwardFilterConvolution() local
295 int64_t padding_high = backward_conv->window().dimensions(i).padding_high(); in CanonicalizeBackwardInputConvolution() local
363 int64_t padding_high = backward_conv->window().dimensions(i).padding_high(); in CanonicalizeBackwardInputConvolution() local
H A Dcudnn_simplify_padding_test.cc92 void ExpectOnlyPadsOneDim(int64_t dim, int64_t padding_high, in ExpectOnlyPadsOneDim()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dxla_pad_op.cc38 std::vector<int64_t> padding_high; in Compile() local
H A Dmatrix_diag_ops.cc208 int padding_high = 0; in SetMatrixDiag() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc372 int64_t padding_high = wd.padding_high(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
670 int64_t padding_high = wd.padding_high(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
H A Dtf_to_xla_attribute_utils.cc46 Value &padding_high) { in GetSamePaddingValues()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto489 int64 padding_high = 4; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc3851 SmallVector<int64_t, 4> padding_low, padding_high, padding_interm; in matchAndRewrite() local