Lines Matching refs:group_input_channels

53     size_t group_input_channels,  in create_deconvolution2d_nhwc()  argument
114 if (group_input_channels == 0) { in create_deconvolution2d_nhwc()
117 xnn_operator_type_to_string(operator_type), group_input_channels); in create_deconvolution2d_nhwc()
128 const size_t input_channels = groups * group_input_channels; in create_deconvolution2d_nhwc()
134 input_pixel_stride, groups, group_input_channels); in create_deconvolution2d_nhwc()
174 const uint32_t k_stride = round_up_po2(group_input_channels, kr); in create_deconvolution2d_nhwc()
218 groups, group_output_channels, kernel_size, group_input_channels, in create_deconvolution2d_nhwc()
225 groups, group_output_channels, kernel_height, kernel_width, group_input_channels, in create_deconvolution2d_nhwc()
257 deconvolution_op->group_input_channels = group_input_channels; in create_deconvolution2d_nhwc()
313 size_t group_input_channels, in xnn_create_deconvolution2d_nhwc_qu8() argument
379 groups, group_input_channels, group_output_channels, in xnn_create_deconvolution2d_nhwc_qu8()
406 size_t group_input_channels, in xnn_create_deconvolution2d_nhwc_f32() argument
457 groups, group_input_channels, group_output_channels, in xnn_create_deconvolution2d_nhwc_f32()
519 const size_t group_input_channels = deconvolution_op->group_input_channels; in setup_conv_path() local
523 …(round_up_po2(group_input_channels, deconvolution_op->ukernel.igemm.kr) * kernel_size << log2_filt… in setup_conv_path()
527 .kc = group_input_channels << log2_input_element_size, in setup_conv_path()
536 .ga_stride = group_input_channels << log2_input_element_size, in setup_conv_path()
678 const size_t group_input_channels = deconvolution_op->group_input_channels; in setup_subconv2d_path() local
683 (round_up_po2(group_input_channels, kr) * kernel_size << log2_filter_element_size); in setup_subconv2d_path()
687 .kc = group_input_channels << log2_input_element_size, in setup_subconv2d_path()
694 .ga_stride = group_input_channels << log2_input_element_size, in setup_subconv2d_path()
706 .kc = group_input_channels << log2_input_element_size, in setup_subconv2d_path()
712 .ga_stride = group_input_channels << log2_input_element_size, in setup_subconv2d_path()