Searched refs:tiled_output_size (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| D | indirection.h | 24 size_t tiled_output_size);
33 size_t tiled_output_size);
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| D | indirection.c | 20 size_t tiled_output_size) { in pytorch_qnnp_indirection_init_conv3d() argument
55 for (size_t output_tile_start = 0; output_tile_start < tiled_output_size; in pytorch_qnnp_indirection_init_conv3d()
84 tiled_output_size * kernel_size + in pytorch_qnnp_indirection_init_conv3d()
106 tiled_output_size * kernel_size + in pytorch_qnnp_indirection_init_conv3d()
120 tiled_output_size * kernel_size + in pytorch_qnnp_indirection_init_conv3d()
342 size_t tiled_output_size) { in pytorch_qnnp_indirection_init_deconv2d() argument
368 for (size_t output_tile_start = 0; output_tile_start < tiled_output_size; in pytorch_qnnp_indirection_init_deconv2d()
387 tiled_output_size * kernel_size + in pytorch_qnnp_indirection_init_deconv2d()
|
| D | deconvolution.c | 282 const size_t tiled_output_size = round_up(output_size, output_tile_size); in pytorch_qnnp_setup_deconvolution2d_nhwc_q8() local
284 sizeof(void*) * batch_size * groups * tiled_output_size * kernel_size; in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
297 deconvolution, output_tile_size, tiled_output_size); in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
|
| D | convolution.c | 768 const size_t tiled_output_size = round_up(output_size, output_tile_size); in pytorch_qnnp_setup_convolution_ndhwc_q8() local
770 sizeof(void*) * batch_size * groups * tiled_output_size * kernel_size; in pytorch_qnnp_setup_convolution_ndhwc_q8()
782 convolution, output_tile_size, tiled_output_size); in pytorch_qnnp_setup_convolution_ndhwc_q8()
|
| /external/XNNPACK/src/ |
| D | indirection.c | 44 const size_t tiled_output_size = round_up(output_size, output_tile_size); in xnn_indirection_init_conv2d() local
49 …for (size_t output_tile_start = 0; output_tile_start < tiled_output_size; output_tile_start += out… in xnn_indirection_init_conv2d()
104 const size_t tiled_output_size = round_up(output_size, output_tile_size); in xnn_indirection_init_deconv2d() local
111 …for (size_t output_tile_start = 0; output_tile_start < tiled_output_size; output_tile_start += out… in xnn_indirection_init_deconv2d()
|
| /external/XNNPACK/src/operators/ |
| D | deconvolution-nhwc.c | 695 const size_t tiled_output_size = round_up(output_size, mr); in setup_conv_path() local
696 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_conv_path()
|
| D | convolution-nhwc.c | 1562 const size_t tiled_output_size = round_up(output_size, mr); in setup_convolution2d_nhwc() local
1563 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_convolution2d_nhwc()
|