21.02
|
Basic function to manage the reshape weights generated from NEConvolutionLayerReshapeWeights. More...
#include <NEGEMMConvolutionLayer.h>
Public Member Functions | |
NEConvolutionLayerReshapeWeightsTransform ()=default | |
Constructor. More... | |
NEConvolutionLayerReshapeWeightsTransform (const NEConvolutionLayerReshapeWeightsTransform &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEConvolutionLayerReshapeWeightsTransform & | operator= (const NEConvolutionLayerReshapeWeightsTransform &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEConvolutionLayerReshapeWeightsTransform (NEConvolutionLayerReshapeWeightsTransform &&)=delete | |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
NEConvolutionLayerReshapeWeightsTransform & | operator= (NEConvolutionLayerReshapeWeightsTransform &&)=delete |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
~NEConvolutionLayerReshapeWeightsTransform ()=default | |
Default destructor. More... | |
void | configure (const ITensor *input, const ITensor *biases) |
void | run () override |
Run the transformation function. More... | |
ITensor * | get_weights () override |
Get a pointer to the transformed weights. More... | |
void | release () override |
Release transformed weights memory. More... | |
uint32_t | uid () override |
Function that returns a unique id of the reshape function. More... | |
bool | is_reshape_run () |
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 the reshape weights generated from NEConvolutionLayerReshapeWeights.
Definition at line 98 of file NEGEMMConvolutionLayer.h.
|
default |
Constructor.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
Prevent instances of this class from being moved (As this class contains non movable objects)
|
default |
Default destructor.
Definition at line 113 of file NEGEMMConvolutionLayer.h.
Referenced by NEGEMMConvolutionLayer::configure().
|
inlineoverridevirtual |
Get a pointer to the transformed weights.
Implements ITransformWeights.
Definition at line 126 of file NEGEMMConvolutionLayer.h.
|
inline |
Definition at line 141 of file NEGEMMConvolutionLayer.h.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
delete |
Prevent instances of this class from being moved (As this class contains non movable objects)
|
inlineoverridevirtual |
Release transformed weights memory.
Implements ITransformWeights.
Definition at line 131 of file NEGEMMConvolutionLayer.h.
|
inlineoverridevirtual |
Run the transformation function.
Implements ITransformWeights.
Definition at line 119 of file NEGEMMConvolutionLayer.h.
|
inlineoverridevirtual |
Function that returns a unique id of the reshape function.
Implements ITransformWeights.
Definition at line 136 of file NEGEMMConvolutionLayer.h.