22.08
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) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
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:169
Descriptors.hpp
src
backends
reference
workloads
LogSoftmax.hpp
Generated on Fri Aug 19 2022 14:38:30 for ArmNN by
1.8.13