理解nvdla设计,首先必须头脑中对数据格式及其内存存放形式有个清晰的认识。
1 输入格式
1)weightdata输入格式:
weight for direct convolution
weight for image input
weight for Winograd convolution
2)activation data输入格式:
feature data format
pixel format (ROI input)
2 输出格式
feature data format
bias data
PReLU data
batch-normalization data
element-wise data
3 精度类型
支持三种精度:int8,int16,fp16
图像在运算前被转成一种类型。featuredata可以是任一一种类型。
引擎支持动态转换类型,需要遵循以下规则:
1)NVDLA convolution pipeline supports precision conversion for image
input mode only.
2)Direct convolution (DC) mode and Winograd convolution mode do not
support precision conversion
3)For image input mode (please see section 6.1.1.4), pipeline allows
conversion from integer to all 3 types. Floating point images can
only be converted to fp16.
3&#x

本文深入探讨NVDLA设计,重点解析其内存中数据格式,包括输入和输出格式的细节。内容涵盖weightdata和activation data的输入格式,如direct convolution、Winograd convolution、feature data等。同时,文章讨论了精度类型转换规则,如int8、int16和fp16的支持情况。此外,还详细介绍了feature data的3D cube组织方式和weight数据的存储顺序。最后,总结了不同硬件层的数据组织格式特点。
最低0.47元/天 解锁文章
2543

被折叠的 条评论
为什么被折叠?



