24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CORE_NEON_KERNELS_NEBATCHNORMALIZATIONLAYERKERNEL_H
25 #define ACL_SRC_CORE_NEON_KERNELS_NEBATCHNORMALIZATIONLAYERKERNEL_H
41 const char *
name()
const override
43 return "NEBatchNormalizationLayerKernel";
109 void configure_non_fused();
111 void configure_fused();
117 using BatchNormFunctionPtr = void (*)(
const Window &
window,
128 BatchNormFunctionPtr _func;
139 #endif // ACL_SRC_CORE_NEON_KERNELS_NEBATCHNORMALIZATIONLAYERKERNEL_H
NEBatchNormalizationLayerKernel()
Default constructor.
~NEBatchNormalizationLayerKernel()=default
Default destructor.
void configure(ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta=nullptr, const ITensor *gamma=nullptr, float epsilon=0.001f, ActivationLayerInfo act_info=ActivationLayerInfo())
Set the input and output tensors.
Interface for CPU tensor.
Activation Layer Information class.
Interface for the batch normalization layer kernel.
Common interface for all kernels implemented in C++.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
const Window & window() const
The maximum window the kernel can be executed on.
const char * name() const override
Name of the kernel.
Information about executing thread and CPU.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
NEBatchNormalizationLayerKernel & operator=(const NEBatchNormalizationLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *mean, const ITensorInfo *var, const ITensorInfo *beta=nullptr, const ITensorInfo *gamma=nullptr, float epsilon=0.001f, ActivationLayerInfo act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of NEBatchNormalizationLaye...