Home
last modified time | relevance | path

Searched defs:pad_width (Results 1 – 25 of 38) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dim2col.cuh29 const int64_t pad_width, in im2col_kernel()
76 const int64_t pad_width, in im2col()
115 const int64_t pad_width, in col2im_device()
170 const int64_t pad_width, in col2im_kernel()
211 const int64_t pad_width, in col2im()
255 const int64_t pad_width, in col2im_batched_kernel()
305 const int64_t pad_width, in col2im_batched()
H A DNaiveConvolutionTranspose2d.cu39 int pad_width, in slow_conv_transpose2d_shape_check()
173 int64_t pad_width = padding[1]; in slow_conv_transpose2d_out_cuda_template() local
358 int64_t pad_width = padding[1]; in slow_conv_transpose2d_backward_out_cuda_template() local
534 int64_t pad_width = padding[1]; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlegacy_reference_ops.h49 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
74 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
90 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
106 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
119 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
152 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
172 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
195 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
212 int dilation_height_factor, int pad_width, int pad_height, in Conv()
238 int dilation_height_factor, int pad_width, int pad_height, in Conv()
[all …]
H A Dconv.h37 const int pad_width = params.padding_values.width; in Conv() local
123 const int pad_width = params.padding_values.width; in Conv() local
217 const int pad_width = params.padding_values.width; in HybridConvPerChannel() local
H A Dtranspose_conv.h35 const int pad_width = params.padding_values.width; in TransposeConv() local
125 const int pad_width = params.padding_values.width; in TransposeConv() local
H A Ddepthwiseconv_uint8.h124 const int pad_width = params.padding_values.width; in Run() local
210 const int pad_width = params.padding_values.width; in RunPerChannel() local
H A Ddepthwiseconv_float.h35 const int pad_width = params.padding_values.width; in DepthwiseConv() local
H A Dconv3d.h50 const int pad_width = params.padding_values.width; in Conv3D() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dim2col_utils.h31 int pad_width, int pad_height, in ExtractPatchIntoBufferColumn()
127 const int pad_width = params.padding_values.width; in DilatedIm2col() local
220 const int pad_width = params.padding_values.width; in Im2col() local
256 const int pad_width = params.padding_values.width; in Im2col() local
291 int pad_depth, int pad_height, int pad_width, // Padding params. in ExtractPatchIntoBufferColumn3D()
383 const int pad_width = params.padding_values.width; in Im2col3D() local
438 const int pad_width = params.padding_values.width; in DilatedIm2col3D() local
H A Dlegacy_optimized_ops.h144 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
174 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
190 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
206 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
242 const int pad_width = params.padding_values.width; in LegacyDepthwiseConvWithRounding() local
287 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
325 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
345 int stride_width, int stride_height, int pad_width, in DepthwiseConv()
368 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
2364 int stride_width, int stride_height, int pad_width, int pad_height, in ExtractPatchIntoBufferColumn()
[all …]
H A Dmultithreaded_conv.h152 const int pad_width = params.padding_values.width; in Conv() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM2d.cpp37 const int64_t pad_width = padding[1]; in compute_columns2d() local
106 int64_t pad_width, in slow_conv2d_shape_check()
233 int64_t pad_width, in slow_conv2d_update_output_frame()
296 int64_t pad_width, in slow_conv2d_backward_update_grad_input_frame()
369 const int64_t pad_width = padding[1]; in slow_conv2d_backward_out_cpu_template() local
492 const int64_t pad_width = padding[1]; in slow_conv2d_backward_weight_out_cpu_template() local
557 const int64_t pad_width = padding[1]; in slow_conv2d_forward_out_cpu() local
H A DConvolutionMM3d.cpp42 const int64_t pad_width = padding[2]; in compute_columns3d() local
123 int64_t pad_width, in slow_conv3d_shape_check()
277 int64_t pad_width, in slow_conv3d_update_output_frame()
325 int64_t pad_width, in slow_conv3d_backward_update_grad_input_frame()
384 const int64_t pad_width = padding[2]; in slow_conv3d_backward_out_cpu_template() local
508 const int64_t pad_width = padding[2]; in slow_conv3d_backward_parameters_out_cpu_template() local
569 const int64_t pad_width = padding[2]; in slow_conv3d_forward_out_cpu() local
H A DNaiveConvolutionTranspose2d.cpp37 int pad_width, in slow_conv_transpose2d_shape_check()
187 int64_t pad_width = padding[1]; in TORCH_META_FUNC() local
259 int64_t pad_width = padding[1]; in slow_conv_transpose2d_out_cpu_template() local
429 int64_t pad_width = padding[1]; in slow_conv_transpose2d_backward_out_cpu_template() local
628 int64_t pad_width = padding[1]; in slow_conv_transpose2d_acc_grad_parameters_cpu() local
H A Dim2col_shape_check.h18 int64_t pad_width, in col2im_shape_check()
146 int64_t pad_width, in im2col_shape_check()
H A DIm2Col.cpp54 int64_t pad_width = padding[1]; in im2col_out_cpu_template() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Ddepthwiseconv_quantized_test.cc174 const int pad_width = params.padding_values.width; in DispatchDepthwiseConvImpl() local
380 const int pad_width = params.padding_values.width; in DispatchDepthwiseConvImpl() local
619 PaddingType padding_type, int pad_width, int pad_height, in TestOneDepthwiseConvWithGivenOutputShift()
756 PaddingType padding_type, int pad_width, int pad_height, in TestOneDepthwiseConvBisectOutputShift()
815 PaddingType padding_type, int pad_width, int pad_height, in TestOneDepthwiseConv()
881 int pad_width, pad_height; in TryTestDepthwiseConv() local
H A Dconv_per_channel_quantized_16x8_test.cc49 const int pad_width = params.padding_values.width; in PickOutputMultiplier() local
146 int* pad_width, int* pad_height, int* stride) { in GenerateValidShapeConfigurations()
198 int pad_width, pad_height; in TryTestOneConvFilter() local
H A Ddepthwiseconv_per_channel_quantized_16x8_test.cc49 const int pad_width = params.padding_values.width; in PickOutputMultiplier() local
145 RuntimeShape* output_shape_inference, int* pad_width, int* pad_height, in GenerateValidShapeConfigurations()
197 int pad_width, pad_height; in TryTestOneDepthwiseConv3x3Filter() local
H A Ddepthwiseconv_per_channel_quantized_test.cc50 const int pad_width = params.padding_values.width; in PickOutputMultiplier() local
193 RuntimeShape* output_shape_inference, int* pad_width, int* pad_height, in GenerateValidShapeConfigurations()
242 int pad_width, pad_height; in TryTestOneDepthwiseConv3x3Filter() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Ddepthwise_conv.h37 const int pad_width = params.padding_values.width; in DepthwiseConvPerChannel() local
136 const int pad_width = params.padding_values.width; in DepthwiseConvPerChannel() local
221 const int pad_width = params.padding_values.width; in DepthwiseConvHybridPerChannel() local
H A Dconv.h39 const int pad_width = params.padding_values.width; in ConvPerChannel() local
151 const int pad_width = params.padding_values.width; in ConvPerChannel() local
H A Dtranspose_conv.h36 const int pad_width = params.padding_values.width; in TransposeConv() local
136 const int pad_width = params.padding_values.width; in TransposeConv() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstring_ngrams_op.cc55 int pad_width = get_pad_width(ngram_width); in get_num_ngrams() local
190 int pad_width = get_pad_width(ngram_width); in CreateNgrams() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Ddepthwise_conv_hybrid.h53 const int pad_width = params.padding_values.width; in DepthwiseConvHybridGeneral() local
289 const int pad_width = params.padding_values.width; in DepthwiseConvHybridWithRounding() local

12