23 #ifndef MMALIB_CNN_CONVOLVEBIAS_ROW_PROCESSWEIGHTS_H_
24 #define MMALIB_CNN_CONVOLVEBIAS_ROW_PROCESSWEIGHTS_H_
26 #include "../../common/MMALIB_types.h"
32 #define MMA_SIZE_BLKS 2
33 #define INDICATOR_BITS 64
35 #define PACKET_ROWS INDICATOR_BITS/__MMA_A_COLS(1)
38 #define PACKET_ROWS_SHIFT 0
39 #define PACKET_ROWS_MASK 0x0ULL
40 #elif PACKET_ROWS == 2
41 #define PACKET_ROWS_SHIFT 1
42 #define PACKET_ROWS_MASK 0x1ULL
43 #elif PACKET_ROWS == 4
44 #define PACKET_ROWS_SHIFT 2
45 #define PACKET_ROWS_MASK 0x3ULL
87 const void *restrict pWeights);
110 const void *restrict pWeights,
111 void *restrict pReorderedWeights);
114 const void *restrict pReorderedWeights,
115 const void *restrict pWeights);
MMALIB_STATUS MMALIB_CNN_convolveBias_row_processWeights_depacketize(const MMALIB_CNN_convolveBias_row_processWeights_Args *pArgs, const void *restrict pReorderedWeights, const void *restrict pWeights)
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 convolu...
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 coefficien...
MMALIB_STATUS_NAME
The enumeration of all status codes.
This structure holds all the input parameters for reordering CNN filter weights.
int32_t pitchA
src0_addr pitchA
int32_t subMChannels
Number of output channels per kernel call.
int8_t packetizeMode
Flag to indicate packetization required.
int32_t No
Number of output channels in total.
int32_t numGroupsPerKernel
Number of groups to be processed in kernel call.
int8_t funcStyle
Variant of the function refer to MMALIB_FUNCTION_STYLE.
int32_t maxSize
Maximum size of reordered subMChannels across all kernel calls. Populated by getMemorySize.
int32_t numInChPerGroup
Number of input channels in one group.
int32_t data_type
Data type must be a member of MMALIB_data_type_e.