[发明专利]张量处理器

摘要

一种张量处理器。张量处理器包括乒乓控制器和与乒乓控制器相连的多个处理引擎。乒乓控制器接收输入张量，根据输入张量的维度信息和权重张量的维度信息计算出要调用的处理引擎的个数并且调用处理引擎组成处理引擎二维阵列。乒乓控制器配置处理引擎二维阵列中的处理引擎之间的连接关系和数据流向。乒乓控制器配置输入张量和权重张量到处理引擎二维阵列。处理引擎二维阵列对输入张量和权重张量进行卷积运算得到输出结果。输出结果被传输到所述乒乓控制器。该张量处理器实现了高速处理输入张量数据且能灵活地应对不同维度的输入张量数据。

主权项

1.一种张量处理器，其包括：乒乓控制器，其接收输入张量；与所述乒乓控制器相连的多个处理引擎；其中，所述乒乓控制器根据所述输入张量的维度计算出要调用的处理引擎的个数并且调用所述个数的处理引擎组成处理引擎二维阵列，配置所述处理引擎二维阵列中的处理引擎之间的连接关系和数据流向，以及配置所述输入张量和权重张量到所述处理引擎二维阵列；所述处理引擎二维阵列对所述输入张量和所述权重张量进行卷积运算得到输出结果，所述输出结果被传输到所述乒乓控制器。