Home
last modified time | relevance | path

Searched refs:num_output_channel_blocks (Results 1 – 2 of 2) sorted by relevance

/external/XNNPACK/src/operators/
Dconvolution-nchw.c263 size_t num_output_channel_blocks = group_output_channels; in xnn_create_convolution2d_nchw_f32() local
271 num_output_channel_blocks = num_output_channel_blocks / 4 + num_output_channel_blocks % 4; in xnn_create_convolution2d_nchw_f32()
281 num_output_channel_blocks = num_output_channel_blocks / 2 + num_output_channel_blocks % 2; in xnn_create_convolution2d_nchw_f32()
297 const size_t packed_weights_size = num_output_channel_blocks * sizeof(uint32_t) + in xnn_create_convolution2d_nchw_f32()
309 convolution_op->num_output_channel_blocks = num_output_channel_blocks; in xnn_create_convolution2d_nchw_f32()
314 … int32_t* input_channel_diffs = (int32_t*) (output_channel_nonzeros + num_output_channel_blocks); in xnn_create_convolution2d_nchw_f32()
315 memset(output_channel_nonzeros, 0, num_output_channel_blocks * sizeof(uint32_t)); in xnn_create_convolution2d_nchw_f32()
588 const size_t num_output_channel_blocks = convolution_op->num_output_channel_blocks; in setup_convolution2d_nchw() local
592 convolution_op->num_output_channel_blocks = num_output_channel_blocks; in setup_convolution2d_nchw()
597 … int32_t* input_channel_diffs = (int32_t*) (output_channel_nonzeros + num_output_channel_blocks); in setup_convolution2d_nchw()
/external/XNNPACK/src/xnnpack/
Doperator.h164 size_t num_output_channel_blocks; member