12 class ScopedCpuTensorHandle;
19 std::unique_ptr<ScopedCpuTensorHandle>
m_Mean;
23 std::unique_ptr<ScopedCpuTensorHandle>
m_Beta;
25 std::unique_ptr<ScopedCpuTensorHandle>
m_Gamma;
void ValidateTensorShapesFromInputs() override
This layer represents a batch normalization operation.
std::unique_ptr< ScopedCpuTensorHandle > m_Gamma
A unique pointer to store Gamma values.
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
std::vector< std::reference_wrapper< std::unique_ptr< ScopedCpuTensorHandle > >> ConstantTensors
std::unique_ptr< ScopedCpuTensorHandle > m_Variance
A unique pointer to store Variance values.
~BatchNormalizationLayer()=default
Default destructor.
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
BatchNormalizationLayer * Clone(Graph &graph) const override
void Accept(ILayerVisitor &visitor) const override
std::unique_ptr< ScopedCpuTensorHandle > m_Mean
A unique pointer to store Mean values.
std::unique_ptr< ScopedCpuTensorHandle > m_Beta
A unique pointer to store Beta values.
BatchNormalizationLayer(const BatchNormalizationDescriptor ¶m, const char *name)
ConstantTensors GetConstantTensorsByRef() override