20.11
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
Tensor.hpp
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
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:153
Descriptors.hpp
src
backends
reference
workloads
LogSoftmax.hpp
Generated on Wed Nov 25 2020 09:59:14 for ArmNN by
1.8.13