ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Softmax.h File Reference
#include "Shape.h"
#include <cstdint>

Go to the source code of this file.

Functions

bool softmaxPrepare (const Shape &input, Shape *output)
 

Function Documentation

◆ softmaxPrepare()

bool softmaxPrepare ( const Shape input,
Shape output 
)

Definition at line 22 of file Softmax.cpp.

23{
24 return genericActivationPrepare(input, output);
25}
bool genericActivationPrepare(const Shape &input, Shape *output)

References genericActivationPrepare().