24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLGEMM_H
25 #define ARM_COMPUTE_CLGEMM_H
40 class CLCompileContext;
53 CLGEMM(std::shared_ptr<IMemoryManager> memory_manager =
nullptr,
IWeightsManager *weights_manager =
nullptr);
136 std::unique_ptr<Impl> _impl;
CLGEMM(std::shared_ptr< IMemoryManager > memory_manager=nullptr, IWeightsManager *weights_manager=nullptr)
Default constructor.
Base class for all functions.
Interface for OpenCL tensor.
~CLGEMM()
Default destructor.
static Status validate(const ITensorInfo *a, const ITensorInfo *b, const ITensorInfo *c, const ITensorInfo *output, float alpha, float beta, const GEMMInfo &gemm_info=GEMMInfo())
Static function to check if given info will lead to a valid configuration of CLGEMM.
Basic function to execute GEMM on OpenCL.
void run() override
Run the kernels contained in the function.
CLGEMM & operator=(const CLGEMM &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void configure(const CLCompileContext &compile_context, const ICLTensor *a, const ICLTensor *b, const ICLTensor *c, ICLTensor *output, float alpha, float beta, const GEMMInfo &gemm_info=GEMMInfo())
Initialise the kernel's inputs and output.
void prepare() override
Prepare the function for executing.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
Weights manager interface to handle weights transformations.