[发明专利]一种计算装置及板卡

摘要

本申请提供一种计算装置及板卡，所述计算装置用于执行LSTM运算，所述板卡，所述板卡包括：存储器件、接口装置和控制器件以及神经网络芯片，所述神经网络芯片包括计算装置，所述存储器件，用于存储数据；所述接口装置，用于实现所述芯片与外部设备之间的数据传输；所述控制器件，用于对所述芯片的状态进行监控。本申请提供的计算装置具有成本低、功耗低的优点。

主权项

1.一种计算装置，其特征在于，所述计算装置用于执行LSTM运算，所述LSTM包括：输入层、隐层、输出层和块block，所述块包括：输入门、输出门和忘记门，所述输入门与输入层连接，所述输出门与输出层连接，所述忘记门与隐层连接，所述计算装置包括：运算单元以及控制器单元；所述运算单元包括：一个主处理电路和从处理电路；所述计算装置用于执行LSTM运算；所述控制器单元，用于获取输入门输入的t时刻输入数据Xit、权值以及忘记门输入的输出数据，所述控制器单元，还用于将输入数据Xit、权值W以及输出数据发送给所述主处理电路；所述主处理电路，用于将输入数据Xit拆分成多个输入数据块，将输出数据拆分成多个输出数据块，将多个输入数据块以及多个输出数据块分发给从处理电路，将所述权值W广播给所述从处理电路；从处理电路，用于将接收到的输入数据块与权值执行乘积运算得到输入中间结果，将接收到的输出数据块与权值执行乘积运算得到输出中间结果，将输入中间结果以及输出中间结果发送给主处理电路；所述主处理电路，还用于将从处理电路的输入中间结果得到部分输出结果，将输出中间结果拼接得到另一部分输出结果，计算部分输出结果和另一部分输出结果的和得到输出门的t时刻的输出结果t。