ArmNN
 24.02
LogSoftmax.hpp
Go to the documentation of this file.
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
Descriptors.hpp
BaseIterator.hpp
armnn::LogSoftmax
void LogSoftmax(Decoder< float > &input, Encoder< float > &output, const TensorInfo &inputInfo, const LogSoftmaxDescriptor &descriptor)
Definition: LogSoftmax.cpp:29
armnn::LogSoftmaxDescriptor
SoftmaxDescriptor LogSoftmaxDescriptor
A LogSoftmaxDescriptor for the LogSoftmaxLayer.
Definition: Descriptors.hpp:196
Tensor.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6