Home
last modified time | relevance | path

Searched defs:pad_height (Results 1 – 25 of 35) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/cuda/
Dim2col.cuh28 const int64_t pad_height, in im2col_kernel()
75 const int64_t pad_height, in im2col()
114 const int64_t pad_height, in col2im_device()
169 const int64_t pad_height, in col2im_kernel()
210 const int64_t pad_height, in col2im()
254 const int64_t pad_height, in col2im_batched_kernel()
304 const int64_t pad_height, in col2im_batched()
DNaiveConvolutionTranspose2d.cu38 int pad_height, in slow_conv_transpose2d_shape_check()
172 int64_t pad_height = padding[0]; in slow_conv_transpose2d_out_cuda_template() local
357 int64_t pad_height = padding[0]; in slow_conv_transpose2d_backward_out_cuda_template() local
533 int64_t pad_height = padding[0]; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
DIm2Col.cu57 int64_t pad_height = padding[0]; in im2col_out_cuda_template() local
/external/pytorch/aten/src/ATen/native/
Dim2col_shape_check.h17 int64_t pad_height, in col2im_shape_check()
145 int64_t pad_height, in im2col_shape_check()
DConvolutionMM2d.cpp36 const int64_t pad_height = padding[0]; in compute_columns2d() local
105 int64_t pad_height, in slow_conv2d_shape_check()
232 int64_t pad_height, in slow_conv2d_update_output_frame()
295 int64_t pad_height, in slow_conv2d_backward_update_grad_input_frame()
368 const int64_t pad_height = padding[0]; in slow_conv2d_backward_out_cpu_template() local
491 const int64_t pad_height = padding[0]; in slow_conv2d_backward_weight_out_cpu_template() local
556 const int64_t pad_height = padding[0]; in slow_conv2d_forward_out_cpu() local
DNaiveConvolutionTranspose2d.cpp36 int pad_height, in slow_conv_transpose2d_shape_check()
186 int64_t pad_height = padding[0]; in TORCH_META_FUNC() local
258 int64_t pad_height = padding[0]; in slow_conv_transpose2d_out_cpu_template() local
428 int64_t pad_height = padding[0]; in slow_conv_transpose2d_backward_out_cpu_template() local
627 int64_t pad_height = padding[0]; in slow_conv_transpose2d_acc_grad_parameters_cpu() local
DConvolutionMM3d.cpp41 const int64_t pad_height = padding[1]; in compute_columns3d() local
122 int64_t pad_height, in slow_conv3d_shape_check()
276 int64_t pad_height, in slow_conv3d_update_output_frame()
324 int64_t pad_height, in slow_conv3d_backward_update_grad_input_frame()
383 const int64_t pad_height = padding[1]; in slow_conv3d_backward_out_cpu_template() local
507 const int64_t pad_height = padding[1]; in slow_conv3d_backward_parameters_out_cpu_template() local
568 const int64_t pad_height = padding[1]; in slow_conv3d_forward_out_cpu() local
DIm2Col.cpp53 int64_t pad_height = padding[0]; in im2col_out_cpu_template() local
DCol2Im.cpp113 int64_t pad_height = padding[0]; in col2im_out_cpu_template() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h31 int pad_width, int pad_height, in ExtractPatchIntoBufferColumn()
128 const int pad_height = params.padding_values.height; in DilatedIm2col() local
221 const int pad_height = params.padding_values.height; in Im2col() local
257 const int pad_height = params.padding_values.height; in Im2col() local
291 int pad_depth, int pad_height, int pad_width, // Padding params. in ExtractPatchIntoBufferColumn3D()
384 const int pad_height = params.padding_values.height; in Im2col3D() local
439 const int pad_height = params.padding_values.height; in DilatedIm2col3D() local
Dlegacy_optimized_ops.h144 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
175 int pad_height, int depth_multiplier, in DepthwiseConv()
191 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv()
206 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
243 const int pad_height = params.padding_values.height; in LegacyDepthwiseConvWithRounding() local
287 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
326 int pad_height, int depth_multiplier, in DepthwiseConv()
346 int pad_height, int depth_multiplier, int32 output_offset, 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 …]
Dmultithreaded_conv.h153 const int pad_height = params.padding_values.height; in Conv() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h49 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
75 int pad_height, int depth_multiplier, in DepthwiseConv()
91 int pad_height, int depth_multiplier, float* output_data, 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()
153 int pad_height, int depth_multiplier, in DepthwiseConv()
173 int pad_height, int depth_multiplier, int32 output_offset, 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 …]
Dconv.h38 const int pad_height = params.padding_values.height; in Conv() local
124 const int pad_height = params.padding_values.height; in Conv() local
218 const int pad_height = params.padding_values.height; in HybridConvPerChannel() local
Dtranspose_conv.h36 const int pad_height = params.padding_values.height; in TransposeConv() local
126 const int pad_height = params.padding_values.height; in TransposeConv() local
Ddepthwiseconv_uint8.h125 const int pad_height = params.padding_values.height; in Run() local
211 const int pad_height = params.padding_values.height; in RunPerChannel() local
Ddepthwiseconv_float.h36 const int pad_height = params.padding_values.height; in DepthwiseConv() local
Dconv3d.h51 const int pad_height = params.padding_values.height; in Conv3D() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dconv_per_channel_quantized_16x8_test.cc50 const int pad_height = params.padding_values.height; in PickOutputMultiplier() local
146 int* pad_width, int* pad_height, int* stride) { in GenerateValidShapeConfigurations()
198 int pad_width, pad_height; in TryTestOneConvFilter() local
Ddepthwiseconv_per_channel_quantized_16x8_test.cc50 const int pad_height = params.padding_values.height; 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
Ddepthwiseconv_per_channel_quantized_test.cc51 const int pad_height = params.padding_values.height; 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
Ddepthwiseconv_quantized_test.cc175 const int pad_height = params.padding_values.height; in DispatchDepthwiseConvImpl() local
381 const int pad_height = params.padding_values.height; 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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h38 const int pad_height = params.padding_values.height; in DepthwiseConvPerChannel() local
137 const int pad_height = params.padding_values.height; in DepthwiseConvPerChannel() local
222 const int pad_height = params.padding_values.height; in DepthwiseConvHybridPerChannel() local
Dconv.h40 const int pad_height = params.padding_values.height; in ConvPerChannel() local
152 const int pad_height = params.padding_values.height; in ConvPerChannel() local
Dtranspose_conv.h37 const int pad_height = params.padding_values.height; in TransposeConv() local
137 const int pad_height = params.padding_values.height; in TransposeConv() local

12