Home
last modified time | relevance | path

Searched defs:padding_low (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_padding_legalization.cc229 int64_t padding_low = backward_conv->window().dimensions(i).padding_low(); in CanonicalizeBackwardFilterConvolution() local
294 int64_t padding_low = backward_conv->window().dimensions(i).padding_low(); in CanonicalizeBackwardInputConvolution() local
362 int64_t padding_low = backward_conv->window().dimensions(i).padding_low(); in CanonicalizeBackwardInputConvolution() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dxla_pad_op.cc37 std::vector<int64_t> padding_low; in Compile() local
H A Dmatrix_diag_ops.cc207 int padding_low = length_to_pad_to - max_diag_len; in SetMatrixDiag() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc371 int64_t padding_low = wd.padding_low(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
669 int64_t padding_low = wd.padding_low(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
H A Dtf_to_xla_attribute_utils.cc45 int64_t stride, Value &padding_low, in GetSamePaddingValues()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla_data.proto484 int64 padding_low = 3; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc1051 Type shape_scalar_type, Value *padding_low, in GetPaddingValues()
3851 SmallVector<int64_t, 4> padding_low, padding_high, padding_interm; in matchAndRewrite() local