24.02.1
|
Basic function to manage CLConvertFullyConnectedWeights. More...
#include <CLConvertFullyConnectedWeights.h>
Public Member Functions | |
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, const TensorShape &original_input_shape, DataLayout data_layout) |
Configures the CLConvertFullyConnectedWeights function. More... | |
void | configure (const CLCompileContext &compile_context, const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_layout) |
Configures the CLConvertFullyConnectedWeights function. More... | |
Public Member Functions inherited from ITransformWeights | |
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 CLConvertFullyConnectedWeights.
Definition at line 112 of file CLConvertFullyConnectedWeights.h.
|
inline |
Configures the CLConvertFullyConnectedWeights function.
[in] | compile_context | The compile context to be used. |
[in] | input | Source weights tensor info to convert. Data type supported: All. |
[in] | original_input_shape | Shape of the original input tensor (the one entering fully connected layer). |
[in] | data_layout | The data layout the weights have been trained in. |
Definition at line 157 of file CLConvertFullyConnectedWeights.h.
References CLConvertFullyConnectedWeights::configure(), arm_compute::cpu::data_layout, and arm_compute::test::validation::input.
|
inline |
Configures the CLConvertFullyConnectedWeights function.
[in] | input | Source weights tensor info to convert. Data type supported: All. |
[in] | original_input_shape | Shape of the original input tensor (the one entering fully connected layer). |
[in] | data_layout | The data layout the weights have been trained in. |
Definition at line 146 of file CLConvertFullyConnectedWeights.h.
References arm_compute::cpu::data_layout, CLKernelLibrary::get(), and arm_compute::test::validation::input.
|
inlineoverridevirtual |
Get a pointer to the transformed weights.
Implements ITransformWeights.
Definition at line 130 of file CLConvertFullyConnectedWeights.h.
|
inlineoverridevirtual |
Release transformed weights memory.
Implements ITransformWeights.
Definition at line 124 of file CLConvertFullyConnectedWeights.h.
References CLTensor::allocator(), and CLTensorAllocator::free().
|
inlineoverridevirtual |
Run the transformation function.
Implements ITransformWeights.
Definition at line 116 of file CLConvertFullyConnectedWeights.h.
References CLTensorAllocator::allocate(), CLTensor::allocator(), and CLConvertFullyConnectedWeights::run().
|
inlineoverridevirtual |
Function that returns a unique id of the reshape function.
Implements ITransformWeights.
Definition at line 136 of file CLConvertFullyConnectedWeights.h.