| 187 | |
| 188 | template <typename T, typename std::enable_if<is_floating_point<T>::value, int>::type> |
| 189 | SimpleTensor<T> softmax_layer(const SimpleTensor<T> &src, float beta, int32_t axis, bool is_log) |
| 190 | { |
| 191 | if (std::is_same<T, bfloat16>::value) |
| 192 | { |
| 193 | return softmax_layer_bfloat16<T>(src, beta, axis, is_log); |
| 194 | } |
| 195 | else |
| 196 | { |
| 197 | return softmax_layer_generic<T>(src, beta, axis, is_log); |
| 198 | } |
| 199 | } |
| 200 | |
| 201 | template <typename T, |
| 202 | typename std::enable_if<std::is_same<T, uint8_t>::value || std::is_same<T, int8_t>::value, int>::type> |
no test coverage detected