Home
last modified time | relevance | path

Searched defs:padded_input (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DActivation.cpp67 Tensor padded_input = mobile::allocate_padded_contiguous_if_needed( in hardswish() local
81 Tensor padded_input = mobile::allocate_padded_contiguous_if_needed( in hardswish_() local
H A DLinear.cpp122 const Tensor padded_input = mobile::allocate_padded_contiguous_if_needed( in run() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dpooling_ops_common.cc308 Tensor padded_input; in Compute() local
604 Tensor padded_input; in Compute() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/utils/
H A DFactory.cpp55 Tensor padded_input = empty_with_tail_padding( in allocate_padded_contiguous_if_needed() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dslice_op.cc171 auto padded_input = in Compile() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLGEMMDeconvolutionLayer.cpp132 …const bool padded_input = deconv_info.pad_bottom() > 0 || deconv_info.pad_left() > 0 || deconv_inf… in validate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_padding_legalization.cc259 HloInstruction* padded_input = in CanonicalizeBackwardFilterConvolution() local
/aosp_15_r20/system/keymint/boringssl/src/
H A Drsa.rs325 let padded_input = zero_pad_left(&self.pending_input, self.max_size)?; in finish() localVariable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolution.cpp1050 auto padded_input = at::constant_pad_nd_symint(input, pad_nd, 0); in convolution_same() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_test.cc9119 auto padded_input = in TEST_F() local