23 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.
m_DataLayout);
24 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.
m_DataLayout);
26 return arm_compute::NEInstanceNormalizationLayer::validate(&aclInputInfo,
44 input.info()->set_data_layout(aclDataLayout);
45 output.info()->set_data_layout(aclDataLayout);
47 m_Layer.configure(&input,
void Execute() const override
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
LayerDescriptor m_Parameters
float m_Gamma
Gamma, the scale scalar value applied for the normalized tensor. Defaults to 1.0. ...
float m_Beta
Beta, the offset scalar value applied for the normalized tensor. Defaults to 1.0. ...
NeonInstanceNormalizationWorkload(const InstanceNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
const InstanceNormalizationQueueDescriptor m_Data
arm_compute::Status NeonInstanceNormalizationWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const InstanceNormalizationDescriptor &descriptor)
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
float m_Eps
Epsilon, small scalar value added to variance to avoid dividing by zero. Defaults to 1e-12f...
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs