Lines Matching refs:group_channels
24 size_t group_channels, in create_channel_shuffle_nc() argument
49 if (group_channels == 0) { in create_channel_shuffle_nc()
52 xnn_operator_type_to_string(operator_type), group_channels); in create_channel_shuffle_nc()
56 const size_t channels = groups * group_channels; in create_channel_shuffle_nc()
61 xnn_operator_type_to_string(operator_type), input_stride, groups, group_channels); in create_channel_shuffle_nc()
69 xnn_operator_type_to_string(operator_type), output_stride, groups, group_channels); in create_channel_shuffle_nc()
84 channel_shuffle_op->group_channels = group_channels; in create_channel_shuffle_nc()
103 size_t group_channels, in xnn_create_channel_shuffle_nc_x8() argument
111 group_channels, in xnn_create_channel_shuffle_nc_x8()
121 size_t group_channels, in xnn_create_channel_shuffle_nc_x32() argument
129 group_channels, in xnn_create_channel_shuffle_nc_x32()
168 .n = channel_shuffle_op->group_channels << log2_element_size, in setup_channel_shuffle_nc()