PDF《一图理解KPU工作原理》

卷积 Convolution 池化 Pooling 批归一化 Batch Normalization 激活 Activate 源数据 Source Data AI内存 AI Memory 2MByte 通用内存 General Memory 卷积核 convolution kernel 批归一化参数 BN parm 激活参数 Activate parm 输入数据 Source Data 输出数据 Dest Data 0x0~data_size 0x200000-data_size~0x200000 每层交换一次Switch every layer 目标数据 Dest Data 通用内存 General Memory 源数据地址 Source Data Addr 目标数据地址 Dest Data Addr 初始化参数 KPU Init Parm 池化参数 Pooling parm DMA DMA KPU 4K(72KB) 卷积核参数寄存器 Convolution Kernel Parm Register 回调函数 CallBack 单层计算 Single Layer Calculation 可跳过 optional bypass 16段折线 16section polyline 列宽 64pixel Col Width 列宽 64pixel Col Width …… Image 图像 列宽 64pixel Col Width 每个KLU处理64宽度的图像 Every KLU deal with 64pixel width image 输入图像列宽需要补齐至64倍数 Input Image Width need padding to 64X KLU KLU KLU KLU Layer0 Parm Layer1 Parm …… LayerN Parm 神经网络层FIFO NN Layer FIFO KPU 单层计算 Single Layer Cal pop done/next 回调 CallBack All Done 剩余收尾工作 SoftMax Or other work 完整网络任务 Full KPU Task 一图理解KPU工作原理 One Diagram Understand How KPU Work Sipeed MAIX https://sipeed.

com support@sipeed.com Analyze From standalone sdk/demo, and Model Compiler