Compute Library
18.05
|
Basic function to reshape the weights of Fully Connected layer with OpenGL ES. More...
#include <GCFullyConnectedLayer.h>
Public Member Functions | |
void | configure (const IGCTensor *input, IGCTensor *output) |
Set the input and output tensors. More... | |
Public Member Functions inherited from IGCSimpleFunction | |
IGCSimpleFunction () | |
Default constructor. More... | |
void | run () overridefinal |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
virtual void | prepare () |
Prepare the function for executing. More... | |
Basic function to reshape the weights of Fully Connected layer with OpenGL ES.
This function calls the following kernels:
Definition at line 43 of file GCFullyConnectedLayer.h.
Set the input and output tensors.
[in] | input | Weights tensor. The weights must be 2 dimensional. Data types supported: F16/F32. |
[out] | output | Destination tensor which stores the transposed input tensor. Data type supported: Same as input . |