ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Softmax.cpp File Reference
#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)