Lines Matching refs:group_channels
24 size_t group_channels, in create_channel_shuffle_nc() argument
47 if (group_channels == 0) { in create_channel_shuffle_nc()
50 group_channels); in create_channel_shuffle_nc()
54 const size_t channels = groups * group_channels; in create_channel_shuffle_nc()
59 input_stride, groups, group_channels); in create_channel_shuffle_nc()
67 output_stride, groups, group_channels); in create_channel_shuffle_nc()
80 channel_shuffle_op->group_channels = group_channels; in create_channel_shuffle_nc()
100 size_t group_channels, in xnn_create_channel_shuffle_nc_x8() argument
108 group_channels, in xnn_create_channel_shuffle_nc_x8()
118 size_t group_channels, in xnn_create_channel_shuffle_nc_x32() argument
126 group_channels, in xnn_create_channel_shuffle_nc_x32()
164 .n = channel_shuffle_op->group_channels << log2_element_size, in setup_channel_shuffle_nc()