10 #include <arm_compute/runtime/CL/CLTensor.h> 11 #include <arm_compute/runtime/CL/functions/CLBatchNormalizationLayer.h> 17 const TensorInfo& output,
18 const TensorInfo& mean,
19 const TensorInfo& var,
20 const TensorInfo& beta,
21 const TensorInfo& gamma,
22 const BatchNormalizationDescriptor& desc);
33 mutable arm_compute::CLBatchNormalizationLayer m_Layer;
35 std::unique_ptr<arm_compute::CLTensor> m_Mean;
36 std::unique_ptr<arm_compute::CLTensor> m_Variance;
37 std::unique_ptr<arm_compute::CLTensor> m_Gamma;
38 std::unique_ptr<arm_compute::CLTensor> m_Beta;
40 void FreeUnusedTensors();
ClBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
void Execute() const override
arm_compute::Status ClBatchNormalizationValidate(const TensorInfo &input, const TensorInfo &output, const TensorInfo &mean, const TensorInfo &var, const TensorInfo &beta, const TensorInfo &gamma, const BatchNormalizationDescriptor &desc)