ArmNN
 24.02
BatchNormImpl.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "Encoders.hpp"
9 #include "Decoders.hpp"
10 
12 
13 namespace armnn
14 {
15 
16 void BatchNormImpl(const BatchNormalizationQueueDescriptor& data,
17  Decoder<float>& meanIn,
18  Decoder<float>& varIn,
19  Decoder<float>& betaIn,
20  Decoder<float>& gammaIn,
21  Decoder<float>& inputData,
22  Encoder<float>& outputData);
23 
24 } // namespace armnn
WorkloadData.hpp
armnn::BatchNormImpl
void BatchNormImpl(const BatchNormalizationQueueDescriptor &data, Decoder< float > &meanDecoder, Decoder< float > &varianceDecoder, Decoder< float > &betaDecoder, Decoder< float > &gammaDecoder, Decoder< float > &inputDecoder, Encoder< float > &outputEncoder)
Definition: BatchNormImpl.cpp:18
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
Encoders.hpp