37 : _memory_group(
std::move(memory_manager)), _squared_input(), _norm_kernel(), _multiply_kernel(), _border_handler()
46 _memory_group.
manage(&_squared_input);
48 _norm_kernel.
configure(input, &_squared_input, output, norm_info);
49 _multiply_kernel.
configure(input, input, &_squared_input, 1.0f);
Class describing the value of a pixel for any image format.
void dispatch(IGCKernel &kernel, bool flush=true)
Schedule the execution of the passed kernel if possible.
virtual DataType data_type() const =0
Data type used for each element of the tensor.
Normalization Layer Information class.
void memory_barrier()
Defines a barrier ordering memory transactions.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Interface for GLES Compute tensor.
GCNormalizationLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
void configure(const IGCTensor *input1, const IGCTensor *input2, IGCTensor *output, float scale)
Initialise the kernel's input, output and border mode.
void init(const TensorInfo &input, size_t alignment=0)
Initialize a tensor based on the passed TensorInfo.
Copyright (c) 2017-2021 Arm Limited.
void manage(IMemoryManageable *obj) override
Sets a object to be managed by the given memory group.
static GCScheduler & get()
Access the scheduler singleton.
virtual const TensorShape & tensor_shape() const =0
Size for each dimension of the tensor.
void configure(const IGCTensor *tensor, BorderSize border_size, BorderMode border_mode, const PixelValue &constant_border_value=PixelValue())
Initialise the kernel's input, output and border mode.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
void configure(const IGCTensor *input, const IGCTensor *squared_input, IGCTensor *output, NormalizationLayerInfo norm_info)
Set the input and output tensors.
virtual void allocate()=0
Interface to be implemented by the child class to allocate the tensor.
void run() override
Run the kernels contained in the function.
BorderSize border_size() const override
The size of the border for that kernel.
Memory group resources scope handling class.
Interface to enqueue GLES kernels and get/set the GLES CommandQueue.
Store the tensor's metadata.
ITensorAllocator * allocator()
Return a pointer to the tensor's allocator.
void configure(const IGCTensor *input, IGCTensor *output, const NormalizationLayerInfo &norm_info)
Set the input and output tensors.