一种在NPU中减少计算资源占用的方法及装置

摘要

本申请提供了一种在NPU中减少计算资源占用的方法及装置，其中NPU包含了用于卷积运算的脉动计算阵列PE array模块，首先获取平均池化的共享除法系数kernel_size，然后将kernel_size扩充为第一卷积核，再将第一卷积核输入PE array模块，通过PE array模块实现所述平均池化。本申请中并未给平均池化设置独立的模块，而是将其映射到原本主要用于实现卷积运算的PE array模块上，将其转换为一类卷积运算，使原有PE array模块可以直接实现任意的平均池化层计算，从而避免了单独的平均池化模块对硬件计算资源的占用，缓解了硬件计算资源紧缺的情况，也降低了NPU的功耗。