24.02.1
|
#include "src/cpu/kernels/softmax/generic/sve/impl.h"
#include "src/core/NEON/wrapper/intrinsics/intrinsics.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2024 Arm Limited. | |
arm_compute::cpu | |
Functions | |
template<typename ScalarType > | |
void | sve_logits_1d_max (const ITensor *in, ITensor *out, const Window &window) |
TODO: (COMPMID-6505) Similar to Neon(TM), this implementation be converted to a single kernel that performs softmax operation. More... | |
template<typename ScalarType > | |
void | sve_softmax_logits_1d_float (const ITensor *in, const ITensor *max, void *const tmp, ITensor *out, const float beta, bool is_log, const Window &window) |