1 // 2 // Copyright © 2019 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <armnn/Descriptors.hpp> 9 #include <armnn/Tensor.hpp> 10 11 #include "BaseIterator.hpp" 12 13 namespace armnn 14 { 15 16 void LogSoftmax(Decoder<float>& input, 17 Encoder<float>& output, 18 const TensorInfo& inputInfo, 19 const LogSoftmaxDescriptor& descriptor); 20 21 } // namespace armnn 22