17 using namespace armcomputetensorutils;
28 const arm_compute::TensorInfo aclInputInfo =
29 armcomputetensorutils::BuildArmComputeTensorInfo(input, desc.
m_DataLayout);
30 const arm_compute::TensorInfo aclOutputInfo =
31 armcomputetensorutils::BuildArmComputeTensorInfo(output, desc.
m_DataLayout);
32 const arm_compute::TensorInfo aclMeanInfo =
33 armcomputetensorutils::BuildArmComputeTensorInfo(mean, desc.
m_DataLayout);
34 const arm_compute::TensorInfo aclVarInfo =
35 armcomputetensorutils::BuildArmComputeTensorInfo(var, desc.
m_DataLayout);
36 const arm_compute::TensorInfo aclBetaInfo =
37 armcomputetensorutils::BuildArmComputeTensorInfo(beta, desc.
m_DataLayout);
38 const arm_compute::TensorInfo aclGammaInfo =
39 armcomputetensorutils::BuildArmComputeTensorInfo(gamma, desc.
m_DataLayout);
42 activationDescriptor);
44 return arm_compute::CLBatchNormalizationLayer::validate(&aclInputInfo,
58 m_Mean = std::make_unique<arm_compute::CLTensor>();
59 BuildArmComputeTensor(*m_Mean,
m_Data.m_Mean->GetTensorInfo());
61 m_Variance = std::make_unique<arm_compute::CLTensor>();
62 BuildArmComputeTensor(*m_Variance,
m_Data.m_Variance->GetTensorInfo());
64 m_Gamma = std::make_unique<arm_compute::CLTensor>();
65 BuildArmComputeTensor(*m_Gamma,
m_Data.m_Gamma->GetTensorInfo());
67 m_Beta = std::make_unique<arm_compute::CLTensor>();
68 BuildArmComputeTensor(*m_Beta,
m_Data.m_Beta->GetTensorInfo());
76 input.info()->set_data_layout(aclDataLayout);
77 output.info()->set_data_layout(aclDataLayout);
81 m_Layer.configure(&input,
107 void ClBatchNormalizationFloatWorkload::FreeUnusedTensors()
109 FreeTensorIfUnused(m_Mean);
110 FreeTensorIfUnused(m_Variance);
111 FreeTensorIfUnused(m_Gamma);
112 FreeTensorIfUnused(m_Beta);
void InitializeArmComputeClTensorData(arm_compute::CLTensor &clTensor, const ConstCpuTensorHandle *handle)
#define ARMNN_SCOPED_PROFILING_EVENT_CL(name)
void RunClFunction(arm_compute::IFunction &function, const CheckLocation &location)
const QueueDescriptor m_Data
arm_compute::ActivationLayerInfo ConvertAdditionalInfoToAclActivationLayerInfo(const QueueDescriptor &queueDescriptor)
float m_Eps
Value to add to the variance. Used to avoid dividing by zero.
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
Copyright (c) 2020 ARM Limited.
ClBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
An ActivationDescriptor for the ActivationLayer.
void Execute() const override
std::vector< ITensorHandle * > m_Outputs
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.
std::vector< ITensorHandle * > m_Inputs
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
arm_compute::ActivationLayerInfo ConvertActivationDescriptorToAclActivationLayerInfo(const ActivationDescriptor &actDesc)