GemmReluMknkChunkGraph< GEMM, CONCAT, NCHUNK, M, K, N, IS_RELU > Class Template Reference

Multiinstance graph for MxK times NxK that stores weights and biases. Requires NxK weight, NCHUNK%8=0 and N%4=0 Chunks NxK weights by N dimension into NCHUNK chunks. Each instance has max size = 16384 and 4096 bytes respectively. Places maximum of 3x3 tiles, 8 conv tiles surrounding concat tile (max AIE DMA input=8) Padding handled within graph, NPAD or KPAD parameters not used. More...

#include <graph_gemm.h>

Inherits adf::graph.

Collaboration diagram for GemmReluMknkChunkGraph< GEMM, CONCAT, NCHUNK, M, K, N, IS_RELU >:

Public Member Functions
	GemmReluMknkChunkGraph (std::vector< float > weights, std::vector< float > bias, int repeat_cnt=1)

Public Attributes
adf::kernel	gemms [CHUNK_COUNT]
ConcatGraph< CONCAT, float_t, CHUNK_COUNT, 1, NCHUNK, N >	concat_g
adf::port< input >	pin [1]
adf::port< output >	pout [1]

Static Public Attributes
static const int	CHUNK_COUNT = (N + NCHUNK - 1) / NCHUNK

Detailed Description

template<template< int, int, int, int > class GEMM, template< typename, int, int, int, int > class CONCAT, int NCHUNK, int M, int K, int N, int IS_RELU>
class GemmReluMknkChunkGraph< GEMM, CONCAT, NCHUNK, M, K, N, IS_RELU >

<dl class=

<td class=

<td class=

---

The documentation for this class was generated from the following file:

• design/aie_src/graph_gemm.h