24.02.1
|
Go to the documentation of this file.
41 struct CLGEMMLowpOutputStage::Impl
43 const ICLTensor *
src{
nullptr};
44 const ICLTensor *
bias{
nullptr};
45 ICLTensor *
dst{
nullptr};
46 std::unique_ptr<opencl::ClGemmLowpOutputStage> op{
nullptr};
47 ITensorPack run_pack{};
77 _impl->op = std::make_unique<opencl::ClGemmLowpOutputStage>();
78 _impl->op->configure(compile_context,
input->info(),
bias !=
nullptr ?
bias->info() :
nullptr, output->
info(),
93 _impl->op->run(_impl->run_pack);
void configure(const ICLTensor *input, const ICLTensor *bias, ICLTensor *output, const GEMMLowpOutputStageInfo &info)
Initialise the kernel's inputs, output.
SimpleTensor< float > src
GEMMLowp output stage info.
Interface for OpenCL tensor.
static CLKernelLibrary & get()
Access the KernelLibrary singleton.
Manages all the OpenCL kernels compilation and caching, provides accessors for the OpenCL Context.
#define ARM_COMPUTE_ERROR_ON_NULLPTR(...)
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
Basic function to execute GEMMLowpQuantizeDown kernels on CL.
static Status validate(const ITensorInfo *input, const ITensorInfo *bias, const ITensorInfo *output, const GEMMLowpOutputStageInfo &info)
Static function to check if given info will lead to a valid configuration of opencl::kernels::ClGemmL...
CLGEMMLowpOutputStage & operator=(const CLGEMMLowpOutputStage &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void run() override
Run the kernels contained in the function.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
static Status validate(const ITensorInfo *src, const ITensorInfo *bias, const ITensorInfo *dst, const GEMMLowpOutputStageInfo &info)
Static function to check if given info will lead to a valid configuration.
~CLGEMMLowpOutputStage()
Default destructor.