Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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
Dcudnn_simplify_padding_test.cc92 void ExpectOnlyPadsOneDim(int64_t dim, int64_t padding_high, in ExpectOnlyPadsOneDim()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_pad_op.cc38 std::vector<int64_t> padding_high; in Compile() local
Dmatrix_diag_ops.cc208 int padding_high = 0; in SetMatrixDiag() local
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
Dtf_to_xla_attribute_utils.cc46 Value &padding_high) { in GetSamePaddingValues()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc372 int64_t padding_high = wd.padding_high(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
670 int64_t padding_high = wd.padding_high(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto489 int64 padding_high = 4; field
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3851 SmallVector<int64_t, 4> padding_low, padding_high, padding_interm; in matchAndRewrite() local