10 #include <arm_compute/runtime/IFunction.h> 11 #include <arm_compute/runtime/Tensor.h> 19 const TensorInfo& output,
20 const TensorInfo& mean,
21 const TensorInfo& var,
22 const TensorInfo& beta,
23 const TensorInfo& gamma,
24 const BatchNormalizationDescriptor& descriptor);
31 virtual void Execute()
const override;
34 std::unique_ptr<arm_compute::IFunction> m_Layer;
36 std::unique_ptr<arm_compute::Tensor> m_Mean;
37 std::unique_ptr<arm_compute::Tensor> m_Variance;
38 std::unique_ptr<arm_compute::Tensor> m_Gamma;
39 std::unique_ptr<arm_compute::Tensor> m_Beta;
41 void FreeUnusedTensors();
arm_compute::Status NeonBatchNormalizationValidate(const TensorInfo &input, const TensorInfo &output, const TensorInfo &mean, const TensorInfo &var, const TensorInfo &beta, const TensorInfo &gamma, const BatchNormalizationDescriptor &descriptor)
Copyright (c) 2020 ARM Limited.
virtual void Execute() const override
NeonBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
Contains information about inputs and outputs to a layer.