Softmax.cpp File Reference

#include "Softmax.hpp"
#include <armnnUtils/TensorUtils.hpp>
#include <cmath>
#include <vector>

Include dependency graph for Softmax.cpp:

Go to the source code of this file.

Namespaces
namespace	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

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.