1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "BaseIterator.hpp" 9 #include <armnn/Tensor.hpp> 10 11 namespace armnn 12 { 13 14 /// Computes the softmax function on some inputs, into outputs, with a shape given by tensorInfo. 15 void Softmax(Decoder<float>& in, Encoder<float>& out, const TensorInfo& inputTensorInfo, float beta, int axis = -1); 16 17 } //namespace armnn 18