projects / platform / core / ml / nntrainer.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: bf029bd) | patch
[batchnorm] Optimize batch norm forward memory
authorParichay Kapoor <pk.kapoor@samsung.com>
Tue, 28 Sep 2021 10:40:33 +0000 (19:40 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Wed, 6 Oct 2021 08:27:10 +0000 (17:27 +0900)
commit62a3695116767aac050031a5a918b09c740f814e
treea74467388dd036104027e7884f0d0ef996b811cftree | snapshot
parentbf029bd96b351c8873e5f9dfd9c735e2cfa66252commit | diff
[batchnorm] Optimize batch norm forward memory

Reduce the memory comsumption of batch norm for forwarding by re-using
the output tensor as temporary memory than requesting new memory.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
nntrainer/layers/bn_layer.cpp diff | blob | history
Domain: Machine Learning / ML Framework;