Lines Matching refs:group_output_channels
50 size_t group_output_channels, in xnn_create_convolution2d_nchw_f32() argument
105 if (group_output_channels == 0) { in xnn_create_convolution2d_nchw_f32()
109 group_output_channels); in xnn_create_convolution2d_nchw_f32()
166 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
172 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
178 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
184 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
212 for (size_t oc = 0; oc < round_down_po2(group_output_channels, 4); oc += 4) { in xnn_create_convolution2d_nchw_f32()
224 …for (size_t oc = round_down_po2(group_output_channels, 4); oc < round_down_po2(group_output_channe… in xnn_create_convolution2d_nchw_f32()
233 for (size_t oc = round_down_po2(group_output_channels, 2); oc < group_output_channels; oc++) { in xnn_create_convolution2d_nchw_f32()
239 size_t num_output_channel_blocks = group_output_channels; in xnn_create_convolution2d_nchw_f32()
274 …(num_nonzero_blocks * 2) * sizeof(int32_t) + (num_nonzero_values + group_output_channels) * sizeof… in xnn_create_convolution2d_nchw_f32()
286 …t32_t* input_increments = (int32_t*) (nonzero_values + num_nonzero_values + group_output_channels); in xnn_create_convolution2d_nchw_f32()
295 …for (size_t ocb = 0; ocb < round_down_po2(group_output_channels, output_channels_block_size); ocb … in xnn_create_convolution2d_nchw_f32()
332 …for (size_t oc = round_down_po2(group_output_channels, output_channels_block_size); oc < group_out… in xnn_create_convolution2d_nchw_f32()
384 round_up(group_output_channels, xnn_params.f32.hwc2spchw_dconv3x3c3s2.output_channel_tile); in xnn_create_convolution2d_nchw_f32()
394 group_output_channels, in xnn_create_convolution2d_nchw_f32()
412 assert(group_output_channels == 1); in xnn_create_convolution2d_nchw_f32()
450 convolution_op->group_output_channels = group_output_channels; in xnn_create_convolution2d_nchw_f32()
523 const size_t group_output_channels = convolution_op->group_output_channels; in setup_convolution2d_nchw() local
524 const size_t output_neurons = groups * group_output_channels * output_height * output_width; in setup_convolution2d_nchw()
529 output_batch_stride, groups, group_output_channels, output_height, output_width); in setup_convolution2d_nchw()
556 …rements = (int32_t*) (nonzero_values + num_nonzero_values + convolution_op->group_output_channels); in setup_convolution2d_nchw()
578 .n = group_output_channels, in setup_convolution2d_nchw()
629 .output_channels = convolution_op->group_output_channels, in setup_convolution2d_nchw()