23 void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs)
const;
24 std::unique_ptr<ScopedTensorHandle> m_Mean;
25 std::unique_ptr<ScopedTensorHandle> m_Variance;
26 std::unique_ptr<ScopedTensorHandle> m_Beta;
27 std::unique_ptr<ScopedTensorHandle> m_Gamma;
RefBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
Copyright (c) 2021 ARM Limited and Contributors.
void Execute() const override
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
Contains information about TensorInfos of a layer.