24.02
|
#include "BatchNormImpl.hpp"
#include "RefWorkloadUtils.hpp"
#include <armnn/Tensor.hpp>
#include <armnnUtils/DataLayoutIndexed.hpp>
#include <cmath>
Go to the source code of this file.
Namespaces | |
armnn | |
Copyright (c) 2021 ARM Limited and Contributors. | |
Functions | |
void | BatchNormImpl (const BatchNormalizationQueueDescriptor &data, Decoder< float > &meanDecoder, Decoder< float > &varianceDecoder, Decoder< float > &betaDecoder, Decoder< float > &gammaDecoder, Decoder< float > &inputDecoder, Encoder< float > &outputEncoder) |