Home
last modified time | relevance | path

Searched defs:stride_width (Results 1 – 25 of 104) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dconv_test.cc55 const TensorData& filter, const TensorData& output, int stride_width = 2, in BaseConvolutionOpModel()
542 const int stride_width = 1; in TEST_P() local
617 const int stride_width = 1; in TEST_P() local
687 const int stride_width = 1; in TEST_P() local
741 const int stride_width = 1; in TEST_P() local
797 const int stride_width = 1; in TEST_P() local
839 const int stride_width = 1; in TEST_P() local
1087 const int stride_width = 1; in TEST_P() local
1394 const int stride_width = 1; in TEST_P() local
1878 const int stride_width = 1; in TEST_P() local
[all …]
H A Dpadding.h63 int stride_height, int stride_width, int dilation_rate_height, in ComputePaddingHeightWidth()
85 int stride_height, int stride_width, int stride_depth, in ComputePadding3DValues()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dim2col.cuh31 const int64_t stride_width, in im2col_kernel()
78 const int64_t stride_width, in im2col()
117 const int64_t stride_width, in col2im_device()
172 const int64_t stride_width, in col2im_kernel()
213 const int64_t stride_width, in col2im()
257 const int64_t stride_width, in col2im_batched_kernel()
307 const int64_t stride_width, in col2im_batched()
H A DNaiveConvolutionTranspose2d.cu37 int stride_width, in slow_conv_transpose2d_shape_check()
175 int64_t stride_width = stride[1]; in slow_conv_transpose2d_out_cuda_template() local
360 int64_t stride_width = stride[1]; in slow_conv_transpose2d_backward_out_cuda_template() local
536 int64_t stride_width = stride[1]; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
H A DNaiveConvolutionTranspose3d.cu37 int stride_width, in slow_conv_transpose3d_shape_check()
219 int64_t stride_width = stride[2]; in slow_conv_transpose3d_out_cuda_template() local
443 int64_t stride_width = stride[2]; in slow_conv_transpose3d_backward_out_cuda_template() local
651 int64_t stride_width = stride[2]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DConv2D.cpp64 int32_t stride_width, stride_height; member
175 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData()
194 int32_t /*padding_bottom*/, int32_t stride_width, int32_t stride_height, in convNhwc()
223 int32_t /*padding_bottom*/, int32_t stride_width, int32_t stride_height, in convNhwc()
277 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc()
306 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc()
333 int32_t stride_width, int32_t stride_height, int32_t dilation_width_factor, in conv()
DGroupedConv2D.cpp53 int32_t /*padding_bottom*/, int32_t stride_width, int32_t stride_height, in groupedConvFloat32()
111 int32_t /*padding_bottom*/, int32_t stride_width, int32_t stride_height, in groupedConvQuant8()
209 int32_t /*padding_bottom*/, int32_t stride_width, in groupedConvQuant8PerChannel()
294 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvFloat16()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM2d.cpp39 const int64_t stride_width = stride[1]; in compute_columns2d() local
104 int64_t stride_width, in slow_conv2d_shape_check()
231 int64_t stride_width, in slow_conv2d_update_output_frame()
294 int64_t stride_width, in slow_conv2d_backward_update_grad_input_frame()
371 const int64_t stride_width = stride[1]; in slow_conv2d_backward_out_cpu_template() local
494 const int64_t stride_width = stride[1]; in slow_conv2d_backward_weight_out_cpu_template() local
559 const int64_t stride_width = stride[1]; in slow_conv2d_forward_out_cpu() local
H A DConvolutionMM3d.cpp45 const int64_t stride_width = stride[2]; in compute_columns3d() local
120 int64_t stride_width, in slow_conv3d_shape_check()
274 int64_t stride_width, in slow_conv3d_update_output_frame()
322 int64_t stride_width, in slow_conv3d_backward_update_grad_input_frame()
387 const int64_t stride_width = stride[2]; in slow_conv3d_backward_out_cpu_template() local
511 const int64_t stride_width = stride[2]; in slow_conv3d_backward_parameters_out_cpu_template() local
572 const int64_t stride_width = stride[2]; in slow_conv3d_forward_out_cpu() local
H A DNaiveConvolutionTranspose2d.cpp35 int stride_width, in slow_conv_transpose2d_shape_check()
189 int64_t stride_width = stride[1]; in TORCH_META_FUNC() local
261 int64_t stride_width = stride[1]; in slow_conv_transpose2d_out_cpu_template() local
431 int64_t stride_width = stride[1]; in slow_conv_transpose2d_backward_out_cpu_template() local
630 int64_t stride_width = stride[1]; in slow_conv_transpose2d_acc_grad_parameters_cpu() local
H A DNaiveConvolutionTranspose3d.cpp37 int stride_width, in slow_conv_transpose3d_shape_check()
223 int64_t stride_width = stride[2]; in slow_conv_transpose3d_out_cpu_template() local
437 int64_t stride_width = stride[2]; in slow_conv_transpose3d_backward_out_cpu_template() local
642 int64_t stride_width = stride[2]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
H A Dim2col_shape_check.h20 int64_t stride_width) { in col2im_shape_check()
148 int64_t stride_width) { in im2col_shape_check()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dpooling.h41 const int stride_width = params.stride_width; in AveragePool() local
99 const int stride_width = params.stride_width; in AveragePool() local
154 const int stride_width = params.stride_width; in L2Pool() local
208 const int stride_width = params.stride_width; in MaxPool() local
262 const int stride_width = params.stride_width; in MaxPool() local
H A Dlegacy_reference_ops.h47 int stride_width, int stride_height, 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()
117 int stride_width, int stride_height, 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()
211 int stride_width, int stride_height, int dilation_width_factor, in Conv()
236 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
254 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
283 int stride_width, int stride_height, int dilation_width_factor, in Conv()
[all …]
H A Dconv.h33 const int stride_width = params.stride_width; in Conv() local
119 const int stride_width = params.stride_width; in Conv() local
213 const int stride_width = params.stride_width; in HybridConvPerChannel() local
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dmax-pooling-nhwc.c45 uint32_t stride_width, in create_max_pooling2d_nhwc()
345 uint32_t stride_width, in xnn_create_max_pooling2d_nhwc_s8()
385 uint32_t stride_width, in xnn_create_max_pooling2d_nhwc_u8()
425 uint32_t stride_width, in xnn_create_max_pooling2d_nhwc_f32()
479 uint32_t stride_width, in xnn_create_max_pooling2d_nhwc_f16()
H A Ddeconvolution-nhwc.c39 uint32_t stride_width, in create_deconvolution2d_nhwc()
296 uint32_t stride_width, in xnn_create_deconvolution2d_nhwc_qs8()
392 uint32_t stride_width, in xnn_create_deconvolution2d_nhwc_qu8()
490 uint32_t stride_width, in xnn_create_deconvolution2d_nhwc_f16()
585 uint32_t stride_width, in xnn_create_deconvolution2d_nhwc_f32()
864 const size_t stride_width = deconvolution_op->stride_width; in setup_subconv2d_path() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dpooling.h41 const int stride_width = params.stride_width; in AveragePool() local
104 const int stride_width = params.stride_width; in MaxPool() local
159 const int stride_width = params.stride_width; in AveragePool() local
222 const int stride_width = params.stride_width; in MaxPool() local
H A Ddepthwise_conv.h33 const int stride_width = params.stride_width; in DepthwiseConvPerChannel() local
132 const int stride_width = params.stride_width; in DepthwiseConvPerChannel() local
217 const int stride_width = params.stride_width; in DepthwiseConvHybridPerChannel() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dim2col_utils.h30 int stride_width, int stride_height, in ExtractPatchIntoBufferColumn()
123 const int stride_width = params.stride_width; in DilatedIm2col() local
218 const int stride_width = params.stride_width; in Im2col() local
254 const int stride_width = params.stride_width; in Im2col() local
290 int stride_depth, int stride_height, int stride_width, // Stride params. in ExtractPatchIntoBufferColumn3D()
380 const int stride_width = params.stride_width; in Im2col3D() local
H A Dlegacy_optimized_ops.h142 int stride_width, int stride_height, 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()
240 const int stride_width = params.stride_width; in LegacyDepthwiseConvWithRounding() local
285 int stride_width, int stride_height, 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()
2364 int stride_width, int stride_height, int pad_width, int pad_height, in ExtractPatchIntoBufferColumn()
2375 const Dims<4>& filter_dims, int stride_width, in DilatedIm2col()
2396 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width, in Im2col()
[all …]
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c37 const size_t stride_width = op->stride_width; in xnn_indirection_init_conv2d() local
97 const size_t stride_width = op->stride_width; in xnn_indirection_init_deconv2d() local
153 const size_t stride_width = op->stride_width; in xnn_indirection_init_subconv2d() local
219 const size_t stride_width = op->stride_width; in xnn_indirection_init_dwconv2d() local
275 const size_t stride_width = op->stride_width; in xnn_indirection_init_maxpool2d() local
866 size_t stride_width, in xnn_indirection_init_pavgpool2d_f16()
892 size_t stride_width, in xnn_indirection_init_pavgpool2d_f32()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c39 const size_t stride_width = op->stride_width; in pytorch_qnnp_indirection_init_conv3d() local
243 const size_t stride_width = op->stride_width; in pytorch_qnnp_indirection_init_dwconv() local
357 const size_t stride_width = op->stride_width; in pytorch_qnnp_indirection_init_deconv2d() local
422 const size_t stride_width = op->stride_width; in pytorch_qnnp_indirection_init_maxpool2d() local
/aosp_15_r20/external/XNNPACK/bench/
H A Ddeconvolution.cc39 const size_t stride_width = state.range(9); in xnnpack_deconvolution_qu8() local
160 const size_t stride_width = state.range(9); in xnnpack_deconvolution_f32() local
278 const size_t stride_width = state.range(9); in tflite_deconvolution_f32() local
/aosp_15_r20/packages/modules/NeuralNetworks/driver/sample_hidl/
DSampleDriverFloatXNNPACK.cpp342 static V1_3::ErrorStatus CheckConvolutionParams(int32_t stride_width, int32_t stride_height, in CheckConvolutionParams()
362 int32_t stride_width, int32_t stride_height, int32_t dilation_width_factor, in CheckDepthwiseConvolutionParams()
388 static V1_3::ErrorStatus CheckPoolingParams(int32_t stride_width, int32_t stride_height, in CheckPoolingParams()
635 int32_t stride_width, stride_height, filter_width, filter_height, activation; in VisitAveragePool2DNode() local
722 int32_t stride_width, stride_height, activation; in VisitConv2DNode() local
822 int32_t stride_width, stride_height, depth_multiplier, activation; in VisitDepthwiseConv2DNode() local
1048 int32_t stride_width, stride_height, filter_width, filter_height, activation; in VisitMaxPool2DNode() local

12345