![]() |
MMALIB User Guide
|
Go to the source code of this file.
Data Structures | |
struct | MMALIB_CNN_convolveBias_row_processWeights_Args |
This structure holds all the input parameters for reordering CNN filter weights. More... | |
Macros | |
#define | MMA_SIZE_BLKS 2 |
#define | INDICATOR_BITS 64 |
#define | SPARSE_BLKS 2 |
#define | PACKET_ROWS INDICATOR_BITS/__MMA_A_COLS(1) |
Functions | |
int32_t | MMALIB_CNN_convolveBias_row_processWeights_getMemorySize (MMALIB_CNN_convolveBias_row_processWeights_Args *pArgs, const void *restrict pWeights) |
This function returns the amount of memory that needs to be allocated for reordered kernel coefficients needed to support MMALIB_CNN_convolveBias_row_ixX_ixX_oxX. More... | |
MMALIB_STATUS | MMALIB_CNN_convolveBias_row_processWeights_reorder (const MMALIB_CNN_convolveBias_row_processWeights_Args *pArgs, const void *restrict pWeights, void *restrict pReorderedWeights) |
This function takes a set of weights and reorders them for use in computing convolve row flow convolution. More... | |
MMALIB_STATUS | MMALIB_CNN_convolveBias_row_processWeights_depacketize (const MMALIB_CNN_convolveBias_row_processWeights_Args *pArgs, const void *restrict pReorderedWeights, const void *restrict pWeights) |
#define MMA_SIZE_BLKS 2 |
Definition at line 32 of file MMALIB_CNN_convolveBias_row_ixX_ixX_oxX_processWeights.h.
#define INDICATOR_BITS 64 |
Definition at line 33 of file MMALIB_CNN_convolveBias_row_ixX_ixX_oxX_processWeights.h.
#define SPARSE_BLKS 2 |
Definition at line 34 of file MMALIB_CNN_convolveBias_row_ixX_ixX_oxX_processWeights.h.
#define PACKET_ROWS INDICATOR_BITS/__MMA_A_COLS(1) |
Definition at line 35 of file MMALIB_CNN_convolveBias_row_ixX_ixX_oxX_processWeights.h.
int32_t MMALIB_CNN_convolveBias_row_processWeights_getMemorySize | ( | MMALIB_CNN_convolveBias_row_processWeights_Args * | pArgs, |
const void *restrict | pWeights | ||
) |
This function returns the amount of memory that needs to be allocated for reordered kernel coefficients needed to support MMALIB_CNN_convolveBias_row_ixX_ixX_oxX.
[in] | pArgs | : Pointer to the structure containing the required dimensional information |
[in] | pWeights | : Pointer to weights array in natural order |
MMALIB_STATUS MMALIB_CNN_convolveBias_row_processWeights_reorder | ( | const MMALIB_CNN_convolveBias_row_processWeights_Args * | pArgs, |
const void *restrict | pWeights, | ||
void *restrict | pReorderedWeights | ||
) |
This function takes a set of weights and reorders them for use in computing convolve row flow convolution.
The function can receive the kernel weights a priori.
[in] | pArgs | : Pointer to argument structure containing necessary parameters for reordering weights |
[in] | pWeights[] | : Pointer to buffer holding naturally ordered convolution weights |
[out] | pReorderedWeights[] | : Pointer to buffer holding the reordered weights output |
MMALIB_STATUS MMALIB_CNN_convolveBias_row_processWeights_depacketize | ( | const MMALIB_CNN_convolveBias_row_processWeights_Args * | pArgs, |
const void *restrict | pReorderedWeights, | ||
const void *restrict | pWeights | ||
) |