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);
decltype(strategy::transforms) typedef type
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
SimpleTensor< T > softmax_layer_generic(const SimpleTensor< T > &src, float beta, int32_t axis, bool is_log)
SimpleTensor< T > softmax_layer(const SimpleTensor< T > &src, float beta, int32_t axis, bool is_log)