Patent ID: 11954583
Assignee: AMAZON TECHNOLOGIES, INC.
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G

Claim 10:
11. A non-transitory computer readable medium storing instructions that, when executed by one or more hardware processors, cause the one or more hardware processors to perform operations for performing a transposed convolution operation using a systolic array, the operations comprising:
loading a first weight data element of an array of weight data elements into the systolic array;
streaming each input data element of a first subset of input data elements into the systolic array to multiply with the first weight data element to compute first partial sums, the first subset being selected based on coordinates of the first weight data element in a 180-degree rotated version of the array of weight data elements;
loading a second weight data element of the array of weight data elements into the systolic array;
streaming each input data element of a second subset of the input data elements into the systolic array to multiply with the second weight data element to compute second partial sums, the second subset being selected based on coordinates of the second weight data element in the 180-degree rotated version of the array of weight data elements; and
generating an output data array of the transposed convolution operation based on the first partial sums and the second partial sums.