ONE - On-device Neural Engine
|
#include "Softmax.h"
#include "Common.h"
#include "QuantizationHelpers.h"
#include <mir/ShapeRange.h>
#include <mir/Tensor.h>
#include <cmath>
Go to the source code of this file.
Data Structures | |
struct | mir_interpreter::SoftmaxImpl< T > |
struct | mir_interpreter::SoftmaxImpl< uint8_t > |
Namespaces | |
namespace | mir_interpreter |
Functions | |
void | mir_interpreter::Softmax (const mir::TensorVariant &arg, int axis, mir::TensorVariant &result) |