24 #ifndef ARM_COMPUTE_TEST_SOFTMAX_LAYER_H
25 #define ARM_COMPUTE_TEST_SOFTMAX_LAYER_H
38 template <typename T, typename std::enable_if<is_floating_point<T>::value,
int>
::type = 0>
41 template <typename T, typename std::enable_if<is_floating_point<T>::value,
int>
::type = 0>
42 SimpleTensor<T>
softmax_layer(
const SimpleTensor<T> &
src,
float beta, int32_t axis = 0,
bool is_log =
false);
44 template < typename T, typename std::enable_if < std::is_same<T, uint8_t>::value || std::is_same<T, int8_t>::value,
int >
::type = 0 >
45 SimpleTensor<T>
softmax_layer(
const SimpleTensor<T> &
src,
float beta, int32_t axis = 0,
bool is_log =
false);