The patent badge is an abbreviated version of the USPTO patent document. The patent badge does contain a link to the full patent document.
The patent badge is an abbreviated version of the USPTO patent document. The patent badge covers the following: Patent number, Date patent was issued, Date patent was filed, Title of the patent, Applicant, Inventor, Assignee, Attorney firm, Primary examiner, Assistant examiner, CPCs, and Abstract. The patent badge does contain a link to the full patent document (in Adobe Acrobat format, aka pdf). To download or print any patent click here.
Patent No.:
Date of Patent:
Sep. 15, 2026
Filed:
Dec. 16, 2022
Amazon Technologies, Inc., Seattle, WA (US);
Pushkar Ratnalikar, San Jose, CA (US);
Hongbin Zheng, San Jose, CA (US);
Ron Diamant, San Jose, CA (US);
Amith Rajith Mamidala, Cupertino, CA (US);
Amazon Technologies, Inc., Seattle, WA (US);
Abstract
A technique for rematerializing a tensor in a neural network model may include identifying a candidate operator for rematerialization in a representation of a data flow graph for a neural network model. An operator can be identified as a candidate operator if it generates a tensor that is used multiple times in the neural network model and belongs to a set of permitted operators for rematerialization. An operator chain that starts at one or more source operators and ends at the candidate operator is determined. A profitability of rematerializing the operator chain is determined based on a comparison of a memory footprint of source data for the one or more source operators, and a memory footprint of the tensor. If it is determined that rematerializing the operator chain is profitable, the operator chain is inserted in the representation of the data flow graph where the tensor is reused.