12 class ScopedCpuTensorHandle;
19 std::unique_ptr<ScopedCpuTensorHandle>
m_Mean;
23 std::unique_ptr<ScopedCpuTensorHandle>
m_Beta;
25 std::unique_ptr<ScopedCpuTensorHandle>
m_Gamma;
This layer represents a batch normalization operation.
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the BatchNormalization type.
BatchNormalizationLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
std::unique_ptr< ScopedCpuTensorHandle > m_Gamma
A unique pointer to store Gamma values.
std::unique_ptr< ScopedCpuTensorHandle > m_Variance
A unique pointer to store Variance values.
Copyright (c) 2020 ARM Limited.
void Accept(ILayerVisitor &visitor) const override
Apply a visitor to this layer.
std::unique_ptr< ScopedCpuTensorHandle > m_Beta
A unique pointer to store Beta values.
std::unique_ptr< ScopedCpuTensorHandle > m_Mean
A unique pointer to store Mean values.
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of BatchNormalizationLayer.
ConstantTensors GetConstantTensorsByRef() override
Retrieve the handles to the constant values stored by the layer.
std::vector< std::reference_wrapper< std::unique_ptr< ScopedCpuTensorHandle > >> ConstantTensors
~BatchNormalizationLayer()=default
Default destructor.
BatchNormalizationLayer(const BatchNormalizationDescriptor ¶m, const char *name)
Constructor to create a BatchNormalizationLayer.
A BatchNormalizationDescriptor for the BatchNormalizationLayer.