Home
last modified time | relevance | path

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

1234

/external/pytorch/aten/src/ATen/native/cuda/
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()
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
/external/tensorflow/tensorflow/lite/kernels/
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 …]
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()
Dconv3d_test.cc35 int32_t stride_width = 1, int32_t stride_height = 1, in Conv3dOpModel()
54 int32_t stride_depth = 1, int32_t stride_width = 1, in Conv3dOpModel()
/external/pytorch/aten/src/ATen/native/
Dim2col_shape_check.h20 int64_t stride_width) { in col2im_shape_check()
148 int64_t stride_width) { in im2col_shape_check()
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
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
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
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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
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
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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
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
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
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 …]
/external/XNNPACK/src/operators/
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()
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
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
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
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Ddeconvolution.c46 uint32_t stride_width, in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
261 const size_t stride_width = deconvolution->stride_width; in pytorch_qnnp_setup_deconvolution2d_nhwc_q8() local
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
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_pool_with_argmax_test.cc37 int stride_width, int filter_height, in MaxpoolingWithArgMaxOpModel()
69 std::vector<uint8_t> CreateCustomOptions(int stride_height, int stride_width, in CreateCustomOptions()
/external/XNNPACK/bench/
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
/external/tensorflow/tensorflow/lite/kernels/internal/
Daveragepool_quantized_test.cc70 const int stride_width = UniformRandomInt(1, 10); in CreateDataAndRunAveragePool() local
133 const int stride_width = UniformRandomInt(1, 128); in CreateExtremalDataAndRunAveragePool() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dpooling.h60 const int stride_width = params.stride_width; in MaxPool() local
170 const int stride_width = params.stride_width; in AveragePool() local
/external/ComputeLibrary/src/core/CPP/kernels/
DCPPUpsampleKernel.cpp76 const int stride_width = _info.stride().first; in run() local

1234