|
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > |
SimpleTensor< T > | softmax_layer_generic (const SimpleTensor< T > &src, float beta, int32_t axis, bool is_log) |
|
template SimpleTensor< float > | softmax_layer_generic (const SimpleTensor< float > &src, float beta, int32_t axis, bool is_log) |
|
template SimpleTensor< half > | softmax_layer_generic (const SimpleTensor< half > &src, float beta, int32_t axis, bool is_log) |
|
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > |
SimpleTensor< T > | softmax_layer (const SimpleTensor< T > &src, float beta, int32_t axis, bool is_log) |
|
template SimpleTensor< float > | softmax_layer (const SimpleTensor< float > &src, float beta, int32_t axis, bool is_log) |
|
template SimpleTensor< half > | softmax_layer (const SimpleTensor< half > &src, float beta, int32_t axis, bool is_log) |
|
template SimpleTensor< uint8_t > | softmax_layer (const SimpleTensor< uint8_t > &src, float beta, int32_t axis, bool is_log) |
|
template SimpleTensor< int8_t > | softmax_layer (const SimpleTensor< int8_t > &src, float beta, int32_t axis, bool is_log) |
|