From db33492c56c9d5aaa94aa5a5187de708cbacf23a Mon Sep 17 00:00:00 2001 From: Jihoon Lee Date: Fri, 15 Oct 2021 18:23:51 +0900 Subject: [PATCH] [Fix] tensor pool prerequest bug This patch fix tensor pool prereuqestedrequest bug by that shared_name was wrong **Self evaluation:** 1. Build test: [X]Passed [ ]Failed [ ]Skipped 2. Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: Jihoon Lee --- nntrainer/tensor/manager.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nntrainer/tensor/manager.cpp b/nntrainer/tensor/manager.cpp index ce45a6a2..024dd83a 100644 --- a/nntrainer/tensor/manager.cpp +++ b/nntrainer/tensor/manager.cpp @@ -362,7 +362,7 @@ Manager::requestTensors(const GraphNode &node, grad = tensor_pool.requestPrerequestedTensor( dim, grad_exec_order, tspan, name + Var_Grad::grad_suffix, /// name - shared_name, + shared_name + Var_Grad::grad_suffix, Tensor::Initializer::ZEROS /// tensor initializer ); } -- 2.34.1