ArmNN
NotReleased
|
Go to the source code of this file.
Namespaces | |
armnn | |
Functions | |
void | Softmax (Decoder< float > &in, Encoder< float > &out, const TensorInfo &inputTensorInfo, float beta, int axis) |
Computes the softmax function on some inputs, into outputs, with a shape given by tensorInfo. More... | |