一种在NPU中减少计算资源占用的方法及装置
基本信息
申请号 | CN202011114887.9 | 申请日 | - |
公开(公告)号 | CN111931927B | 公开(公告)日 | 2020-11-13 |
申请公布号 | CN111931927B | 申请公布日 | 2020-11-13 |
分类号 | G06N3/063(2006.01)I; | 分类 | 计算;推算;计数; |
发明人 | 戴舒诣;范名超 | 申请(专利权)人 | 翱捷智能科技(上海)有限公司 |
代理机构 | 北京清大紫荆知识产权代理有限公司 | 代理人 | 冯振华 |
地址 | 200120上海市浦东新区南汇新城镇环湖西二路888号C楼 | ||
法律状态 | - |
摘要
摘要 | 本申请提供了一种在NPU中减少计算资源占用的方法及装置,其中NPU包含了用于卷积运算的脉动计算阵列PE array模块,首先获取平均池化的共享除法系数kernel_size,然后将kernel_size扩充为第一卷积核,再将第一卷积核输入PE array模块,通过PE array模块实现所述平均池化。本申请中并未给平均池化设置独立的模块,而是将其映射到原本主要用于实现卷积运算的PE array模块上,将其转换为一类卷积运算,使原有PE array模块可以直接实现任意的平均池化层计算,从而避免了单独的平均池化模块对硬件计算资源的占用,缓解了硬件计算资源紧缺的情况,也降低了NPU的功耗。 |
