Compute Library
 21.08
SoftmaxLayer.cpp File Reference

Go to the source code of this file.

Namespaces

 arm_compute
 Copyright (c) 2017-2021 Arm Limited.
 
 arm_compute::test
 
 arm_compute::test::validation
 
 arm_compute::test::validation::reference
 

Functions

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)