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