23 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.
m_DataLayout);
24 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.
m_DataLayout);
26 arm_compute::NormalizationLayerInfo layerInfo = BuildArmComputeNormalizationLayerInfo(descriptor);
28 return arm_compute::CLNormalizationLayer::validate(&aclInputInfo, &aclOutputInfo, layerInfo);
41 input.info()->set_data_layout(aclDataLayout);
42 output.info()->set_data_layout(aclDataLayout);
44 arm_compute::NormalizationLayerInfo normalizationInfo = BuildArmComputeNormalizationLayerInfo(
m_Data.m_Parameters);
46 m_NormalizationLayer.configure(&input, &output, normalizationInfo);
void RunClFunction(arm_compute::IFunction &function, const CheckLocation &location)
A NormalizationDescriptor for the NormalizationLayer.
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
const QueueDescriptor m_Data
void Execute() const override
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
ClNormalizationFloatWorkload(const NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
#define ARMNN_SCOPED_PROFILING_EVENT_CL(name)
arm_compute::Status ClNormalizationWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const NormalizationDescriptor &descriptor)