22 #ifndef MMALIB_CNN_CONVOLVE_COL_SMALLNO_HIGHPRECISION_REORDERWEIGHTS_H_
23 #define MMALIB_CNN_CONVOLVE_COL_SMALLNO_HIGHPRECISION_REORDERWEIGHTS_H_
25 #include "../../common/MMALIB_types.h"
26 #include "../MMALIB_CNN_convolve_col_smallNo_highPrecision_pointwisePost/MMALIB_CNN_convolve_col_smallNo_highPrecision_pointwisePost_reorderWeights.h"
66 typedef enum _MMALIB_reorder_weights_highPrecision_operation_e
183 const void *restrict pWeights,
185 const void *restrict pBias,
187 void *restrict pReorderedWeights);
MMALIB_STATUS MMALIB_CNN_convolve_col_smallNo_highPrecision_reorderWeights_exec(uint32_t operation, const MMALIB_CNN_convolve_col_smallNo_highPrecision_reorderWeights_Args *pArgs, const MMALIB_bufParams3D_t *pWeights_addr, const void *restrict pWeights, const MMALIB_bufParams2D_t *pBias_addr, const void *restrict pBias, const MMALIB_bufParams3D_t *pReorderedWeights_addr, void *restrict pReorderedWeights)
This function takes a set of weights and biases (optional) and reorders the coefficients for improved...
MMALIB_CNN_convolve_col_smallNo_highPrecision_pointwisePost_reorderWeights_Args MMALIB_CNN_convolve_col_smallNo_highPrecision_reorderWeights_Args
MMALIB_reorder_weights_highPrecision_operation_e
Enumeration for different operations for the reorderWeights function.
MMALIB_STATUS MMALIB_CNN_convolve_col_smallNo_highPrecision_reorderWeights_fillBufParams(const MMALIB_CNN_convolve_col_smallNo_highPrecision_reorderWeights_Args *pArgs, MMALIB_bufParams3D_t *pReorderedWeights_addr)
This function populates a structure of dimensional information about the reordered coefficients buffe...
int32_t MMALIB_CNN_convolve_col_smallNo_highPrecision_reorderWeights_getMemorySize(const MMALIB_CNN_convolve_col_smallNo_highPrecision_reorderWeights_Args *pArgs)
This function returns the amount of memory that needs to be allocated for reordered kernel coefficien...
@ HIGHPRECISION_REORDERBIAS
@ HIGHPRECISION_REORDERWEIGHTS
@ HIGHPRECISION_REORDERWEIGHTS_AND_BIAS
MMALIB_STATUS_NAME
The enumeration of all status codes.
This structure holds all the input parameters for reordering CNN filter weights for column-based conv...
This structure holds all the input parameters for reordering CNN filter weights for column-based conv...
A structure for a 2 dimensional buffer descriptor.
A structure for a 3 dimensional buffer descriptor.