43 _norm_kernel.
configure(input, output, mean, var, beta, gamma, epsilon, act_info);
void dispatch(IGCKernel &kernel, bool flush=true)
Schedule the execution of the passed kernel if possible.
Interface for GLES Compute tensor.
void run() override
Run the kernels contained in the function.
void configure(const IGCTensor *input, IGCTensor *output, const IGCTensor *mean, const IGCTensor *var, const IGCTensor *beta, const IGCTensor *gamma, float epsilon, ActivationLayerInfo act_info=ActivationLayerInfo())
Set the input and output tensors.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
static GCScheduler & get()
Access the scheduler singleton.
GCBatchNormalizationLayer()
Default constructor.
void configure(const IGCTensor *input, IGCTensor *output, const IGCTensor *mean, const IGCTensor *var, const IGCTensor *beta=nullptr, const IGCTensor *gamma=nullptr, float epsilon=0.001f, ActivationLayerInfo act_info=ActivationLayerInfo())
Set the input and output tensors.
Interface to enqueue GLES kernels and get/set the GLES CommandQueue.