Home
last modified time | relevance | path

Searched defs:padding_needed (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dconv_grad_size_util.cc45 const int64_t padding_needed = in GetWindowedOutputSize() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dkernel_shape_util.cc48 const int64_t padding_needed = in GetWindowedOutputSizeVerboseV2() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_window_utils.cc118 HloOp padding_needed = Maximum( in GetWindowedOutputSize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
H A Dtf_to_xla_attribute_utils.cc80 Value padding_needed = scalar_sub( in GetSamePaddingValues() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DConvolution.cpp269 uint32_t padding_needed = L_aligned - L; in rearrange_bias() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc1096 Value padding_needed = rewriter.create<arith::SubIOp>( in GetPaddingValues() local