Vector computation unit in a neural network processor
First Claim
Patent Images
1. A circuit for performing neural network computations for a neural network comprising a plurality of layers, the circuit comprising:
- a vector computation unit, wherein the vector computation unit includes;
activation circuitry configured to receive a vector of accumulated values, where the accumulated values correspond to products of a matrix multiplication between an input to a layer of the neural network and a parameter matrix for the layer, and where the activation circuitry is configured to apply a function to each accumulated value to generate a vector of activation values; and
normalization circuitry coupled to the activation circuitry and configured to receive activation values of the vector of activation values and to generate a respective normalized value for each activation value, the normalization circuitry comprising a plurality of normalization register columns, each normalization register column comprising a plurality of normalization registers, and wherein at least one normalization register in each normalization register column is configured to receive a respective activation value for generating at least the respective normalized value for the activation value.
2 Assignments
0 Petitions
Accused Products
Abstract
A circuit for performing neural network computations for a neural network comprising a plurality of layers, the circuit comprising: activation circuitry configured to receive a vector of accumulated values and configured to apply a function to each accumulated value to generate a vector of activation values; and normalization circuitry coupled to the activation circuitry and configured to generate a respective normalized value from each activation value.
50 Citations
16 Claims
-
1. A circuit for performing neural network computations for a neural network comprising a plurality of layers, the circuit comprising:
a vector computation unit, wherein the vector computation unit includes; activation circuitry configured to receive a vector of accumulated values, where the accumulated values correspond to products of a matrix multiplication between an input to a layer of the neural network and a parameter matrix for the layer, and where the activation circuitry is configured to apply a function to each accumulated value to generate a vector of activation values; and normalization circuitry coupled to the activation circuitry and configured to receive activation values of the vector of activation values and to generate a respective normalized value for each activation value, the normalization circuitry comprising a plurality of normalization register columns, each normalization register column comprising a plurality of normalization registers, and wherein at least one normalization register in each normalization register column is configured to receive a respective activation value for generating at least the respective normalized value for the activation value. - View Dependent Claims (2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12)
-
13. A method for performing neural network computations for a neural network comprising a plurality of layers, the method comprising:
-
receiving, at an activation circuit in a vector computation unit and from a systolic array, a vector of accumulated values, where the accumulated values correspond to products of a matrix multiplication between an input to a layer of the neural network and a parameter matrix for the layer; applying, at the activation circuit in the vector computation unit, a function to each accumulated value to generate a vector of activation values; receiving, at a normalization circuit in the vector computation unit, the vector of activation values from the activation circuit, wherein the normalization circuit is coupled to the activation circuit in the vector computation unit, and wherein the normalization circuit comprises multiple normalization register columns, each normalization register column comprising multiple normalization registers, and wherein at least one normalization register in each normalization register column is configured to receive a respective activation value of the vector of activation values; and generating, at the normalization circuit in the vector computation unit, a respective normalized value for each activation value received by the at least one normalization register in a normalization register column. - View Dependent Claims (14, 15, 16)
-
Specification