公开(公告)号：US12175222B1

公开(公告)日：2024-12-24

申请号：US16949958

申请日：2020-11-20

Applicant: Amazon Technologies, Inc.

Inventor： Michael Ray Benfield ,  Hongbin Zheng ,  Thomas Robert Norell

IPC: G06N3/084 ,  G06F8/41 ,  G06N3/04 ,  G06F17/16

Abstract: A computer-implemented method includes generating, based on a representation of a tensor mapping between an input tensor and an output tensor, a list of mappings from elements of the input tensor to elements of the output tensor, and generating groups of mappings from the list of mappings, where each of the groups of mappings corresponds to a respective set of matrix multiplications, a matrix transpose, or both. The computer-implemented method also includes generating a respective expression for each of the groups of mappings and generating code for summing results of the respective expressions, where each respective expression includes the respective set of matrix multiplications, the matrix transpose, or both.