Method and apparatus for allocating memory space for driving neural network
AssigneeSamsung Electronics Co., Ltd.
InventorJoonho Song
Life of this patent
Estimating the term…
Abstract
A method of allocating a memory for driving a neural network including obtaining first capacity information of a space to store an input feature map of a first layer from among the layers of the neural network, and second capacity information of a space to store an output feature map of the first layer, and allocating a first storage space to store the input feature map in the memory based on an initial address value of the memory and the first capacity information and a second storage space to store the output feature map in the memory based on a last address value of the memory and the second capacity information.
Claims15
A method of allocating a memory to layers of a neural network, the method comprising:
- obtaining first capacity information of a space to store an input feature map of a first layer from among the layers of the neural network, and second capacity information of a space to store an output feature map of the first layer;
- and
- allocating a first storage space to store the input feature map in the memory based on an initial address value of the memory and the first capacity information and a second storage space to store the output feature map in the memory based on a last address value of the memory and the second capacity information,
- wherein the first store e space corresponds to a space from the initial address value to a first address value of the memory, and
- wherein the second storage space corresponds to a space from a second address value to the last address value of the memory.
A neural network device comprising:
- a memory;
- and
- a processor configured to drive to execute instructions to drive a neural network to:
- obtain first capacity information of a space to store an input feature map of a first layer from among layers of the neural network, and second capacity information of a space to store an output feature map of the first layer, and
- allocate a first storage space to store the input feature map in the memory based on an initial address value of the memory and the first capacity information and a second storage space to store the output feature map in the memory based on a last address value of the memory and the second capacity information,
- wherein the first storage space corresponds to a space from the initial address value to a first address value of the memo and the second storage space corresponds to a space from a second address value to the last address value of the memory.
Dependencies are inferred from the claim text.