Go to the source code of this file.
|
template<typename T > |
SimpleTensor< T > | channel_shuffle (const SimpleTensor< T > &src, int num_groups) |
|
template SimpleTensor< uint8_t > | channel_shuffle (const SimpleTensor< uint8_t > &src, int num_groups) |
|
template SimpleTensor< uint16_t > | channel_shuffle (const SimpleTensor< uint16_t > &src, int num_groups) |
|
template SimpleTensor< uint32_t > | channel_shuffle (const SimpleTensor< uint32_t > &src, int num_groups) |
|
template SimpleTensor< half > | channel_shuffle (const SimpleTensor< half > &src, int num_groups) |
|
template SimpleTensor< float > | channel_shuffle (const SimpleTensor< float > &src, int num_groups) |
|