24.02.1
|
#include "arm_compute/core/Helpers.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 | sve2_softmax_logits_1d_quantized (const ITensor *in, const ITensor *max, void *const tmp, ITensor *out, float beta, bool is_log, const Window &window) |
TODO: (COMPMID-6505) Similar to Neon(TM), this implementation be converted to a single kernel that performs softmax operation. More... | |