Lines Matching refs:groups
23 size_t groups, in create_channel_shuffle_nc() argument
42 if (groups <= 1) { in create_channel_shuffle_nc()
45 xnn_operator_type_to_string(operator_type), groups); 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()
83 channel_shuffle_op->groups = groups; in create_channel_shuffle_nc()
102 size_t groups, in xnn_create_channel_shuffle_nc_x8() argument
110 groups, in xnn_create_channel_shuffle_nc_x8()
120 size_t groups, in xnn_create_channel_shuffle_nc_x32() argument
128 groups, in xnn_create_channel_shuffle_nc_x32()
162 const size_t groups = channel_shuffle_op->groups; in setup_channel_shuffle_nc() local
169 .m = groups, in setup_channel_shuffle_nc()
173 switch (groups) { in setup_channel_shuffle_nc()