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,
23 const ActivationDescriptor* activationDescriptor =
nullptr);
30 const arm_compute::CLCompileContext& clCompileContext);
36 mutable arm_compute::CLBatchNormalizationLayer m_Layer;
38 std::unique_ptr<arm_compute::CLTensor> m_Mean;
39 std::unique_ptr<arm_compute::CLTensor> m_Variance;
40 std::unique_ptr<arm_compute::CLTensor> m_Gamma;
41 std::unique_ptr<arm_compute::CLTensor> m_Beta;
43 void FreeUnusedTensors();
ClBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Copyright (c) 2021 ARM Limited and Contributors.
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, const ActivationDescriptor *activationDescriptor)
Contains information about inputs and outputs to a layer.