[Manager] Use TensorPool for Gradients
authorParichay Kapoor <pk.kapoor@samsung.com>
Wed, 25 Aug 2021 04:48:46 +0000 (13:48 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 30 Sep 2021 23:13:38 +0000 (08:13 +0900)
commitfdb84ac99c4915e87981f6f01d21147ebfc1745d
treecdc6befb4b33726845dcfdc3b7b5e26f4189d3b7
parent979cb371c19c99cff281374320f302bd9666f30d
[Manager] Use TensorPool for Gradients

Use TensorPool for gradients of the weights.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
jni/Android.mk
nntrainer/graph/network_graph.cpp
nntrainer/layers/layer_context.cpp
nntrainer/layers/layer_context.h
nntrainer/layers/layer_node.h
nntrainer/tensor/manager.cpp
nntrainer/tensor/manager.h