Low latency matrix multiply unit
AssigneeGoogle Inc.
InventorsAndrew Everett Phelps, Norman Paul Jouppi
Life of this patent
Estimating the term…
Loading patent…
AssigneeGoogle Inc.
InventorsAndrew Everett Phelps, Norman Paul Jouppi
Estimating the term…
Methods, systems, and apparatus for a matrix multiply unit implemented as a systolic array of cells are disclosed. Each cell of the matrix multiply includes: a weight matrix register configured to receive a weight input from either a transposed or a non-transposed weight shift register; a transposed weight shift register configured to receive a weight input from a horizontal direction to be stored in the weight matrix register; a non-transposed weight shift register configured to receive a weight input from a vertical direction to be stored in the weight matrix register; and a multiply unit that is coupled to the weight matrix register and configured to multiply the weight input of the weight matrix register with a vector data input in order to obtain a multiplication result.
A processor comprising:
A method performed by a processor, wherein the processor comprises a processing core having an array of cells, a first plurality of shift chains, and a second plurality of shift chains, the method comprising:
A non-transitory computer program product storing instructions that, when executed by a programmable processor comprising a processing core having an array of cells, a first plurality of shift chains, and a second plurality of shift chains, cause the processor to perform operations comprising:
Dependencies are inferred from the claim text.