Searched defs:channel_shuffle (Results 1 – 2 of 2) sorted by relevance
39 SimpleTensor<T> channel_shuffle(const SimpleTensor<T> &src, int num_groups) in channel_shuffle() function
369 struct channel_shuffle_context channel_shuffle; member