21.02
|
Basic function to manage the reshape weights generated from CLGEMMReshapeRHSMatrixKernel. More...
#include <CLGEMM.h>
Public Member Functions | |
CLGEMMReshapeRHSMatrixKernelManaged () | |
Default constructor. More... | |
CLGEMMReshapeRHSMatrixKernelManaged (const CLGEMMReshapeRHSMatrixKernelManaged &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
CLGEMMReshapeRHSMatrixKernelManaged (CLGEMMReshapeRHSMatrixKernelManaged &&)=default | |
Default move constructor. More... | |
CLGEMMReshapeRHSMatrixKernelManaged & | operator= (const CLGEMMReshapeRHSMatrixKernelManaged &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
CLGEMMReshapeRHSMatrixKernelManaged & | operator= (CLGEMMReshapeRHSMatrixKernelManaged &&)=default |
Default move assignment operator. More... | |
~CLGEMMReshapeRHSMatrixKernelManaged () | |
Default desctructor. More... | |
void | run () override |
Run the transformation function. More... | |
void | release () override |
Release transformed weights memory. More... | |
ICLTensor * | get_weights () override |
Get a pointer to the transformed weights. More... | |
uint32_t | uid () override |
Function that returns a unique id of the reshape function. More... | |
void | configure (const ICLTensor *input, GEMMRHSMatrixInfo info) |
Configures the CLGEMMReshapeRHSMatrixKernel kernel. More... | |
void | configure (const CLCompileContext &compile_context, const ICLTensor *input, GEMMRHSMatrixInfo info) |
Configures the CLGEMMReshapeRHSMatrixKernel kernel. More... | |
![]() | |
ITransformWeights ()=default | |
Default Constructor. More... | |
virtual | ~ITransformWeights ()=default |
Default Destructor. More... | |
ITransformWeights (const ITransformWeights &)=delete | |
Prevent instances of this class to be copy constructed. More... | |
ITransformWeights & | operator= (const ITransformWeights &)=delete |
Prevent instances of this class to be copied. More... | |
ITransformWeights (ITransformWeights &&other) | |
Allow instances of this class to be move constructed. More... | |
ITransformWeights & | operator= (ITransformWeights &&other) |
Allow instances of this class to be moved. More... | |
void | increase_refcount () |
Increase the object's refcount. More... | |
int32_t | decrease_refcount () |
Decrease the object's refcount and return the updated value. More... | |
bool | is_reshape_run () |
Function that returns a flag on whether the weights are reshaped or not. More... | |
Basic function to manage the reshape weights generated from CLGEMMReshapeRHSMatrixKernel.
Default constructor.
Definition at line 60 of file CLGEMM.cpp.
References CLGEMMReshapeRHSMatrixKernelManaged::~CLGEMMReshapeRHSMatrixKernelManaged().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
Default move constructor.
|
default |
Default desctructor.
Referenced by CLGEMMReshapeRHSMatrixKernelManaged::CLGEMMReshapeRHSMatrixKernelManaged().
void configure | ( | const ICLTensor * | input, |
GEMMRHSMatrixInfo | info | ||
) |
Configures the CLGEMMReshapeRHSMatrixKernel kernel.
[in] | input | Input tensor. Data types supported: All |
[in] | info | RHS matrix information to be used for reshaping. |
Definition at line 89 of file CLGEMM.cpp.
References CLKernelLibrary::get().
void configure | ( | const CLCompileContext & | compile_context, |
const ICLTensor * | input, | ||
GEMMRHSMatrixInfo | info | ||
) |
Configures the CLGEMMReshapeRHSMatrixKernel kernel.
[in] | compile_context | The compile context to be used. |
[in] | input | Input tensor. Data types supported: All |
[in] | info | RHS matrix information to be used for reshaping. |
Definition at line 94 of file CLGEMM.cpp.
References ARM_COMPUTE_LOG_INFO_MSG_WITH_FORMAT_CORE, arm_compute::auto_init_if_empty(), arm_compute::test::validation::b, ICloneable< T >::clone(), arm_compute::misc::shape_calculator::compute_lhs_reshaped_shape(), arm_compute::misc::shape_calculator::compute_rhs_reshaped_shape(), GEMMKernelInfo::has_pad_y, GEMMKernelInfo::lhs_info, arm_compute::NATIVE_V1, arm_compute::RESHAPED, arm_compute::RESHAPED_ONLY_RHS, arm_compute::RESHAPED_V1, GEMMKernelInfo::rhs_info, arm_compute::cl_gemm::auto_heuristics::select_default_gemm_config_reshaped(), arm_compute::cl_gemm::auto_heuristics::select_default_gemm_config_reshaped_only_rhs(), arm_compute::cl_gemm::auto_heuristics::select_default_gemm_kernel(), arm_compute::cl_gemm::auto_heuristics::select_mlgo_gemm_config_reshaped(), arm_compute::cl_gemm::auto_heuristics::select_mlgo_gemm_config_reshaped_only_rhs(), arm_compute::cl_gemm::auto_heuristics::select_mlgo_gemm_kernel(), arm_compute::to_string(), CLGEMMReshapeLHSMatrixKernel::validate(), CLGEMMReshapeRHSMatrixKernel::validate(), CLGEMMMatrixMultiplyReshapedOnlyRHSKernel::validate(), and CLGEMMMatrixMultiplyReshapedKernel::validate().
|
overridevirtual |
Get a pointer to the transformed weights.
Implements ITransformWeights.
Definition at line 79 of file CLGEMM.cpp.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Default move assignment operator.
|
overridevirtual |
Release transformed weights memory.
Implements ITransformWeights.
Definition at line 74 of file CLGEMM.cpp.
References CLTensor::allocator(), and CLTensorAllocator::free().
|
overridevirtual |
Run the transformation function.
Implements ITransformWeights.
Definition at line 67 of file CLGEMM.cpp.
References CLTensorAllocator::allocate(), CLTensor::allocator(), CLScheduler::enqueue(), and CLScheduler::get().
|
overridevirtual |
Function that returns a unique id of the reshape function.
Implements ITransformWeights.
Definition at line 84 of file CLGEMM.cpp.