Patent ID: 11915147
Assignee: INTERNATIONAL BUSINESS MACHINES CORPORATION
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G

Claim 7:
8. A computer-implemented method, comprising:
training, by a graphics processing unit using a graphics processing unit cache memory of the graphics processing unit, a deep neural network that comprises a set of layers, wherein the training comprises:
determining, by the graphics processing unit, during a forward pass process of training the deep neural network that traverses through the set of layers from a first layer of the set of layers to a last layer of the set of layers, that data from the graphics processing unit cache memory employed for the training by an intermediate layer of the set of layers between the first layer and the last layer, will be employed by the intermediate layer during a backward pass process of training the deep neural network that traverses from the last layer to the first layer; and
transmitting, by the graphics processing unit during the forward pass process, the data to a central processing unit for storage in a central processing unit cache memory of the central processing unit.