projects
/
platform
/
core
/
ml
/
nntrainer.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
59c3dae
)
[Fix] tensor pool prerequest bug
author
Jihoon Lee
<jhoon.it.lee@samsung.com>
Fri, 15 Oct 2021 09:23:51 +0000
(18:23 +0900)
committer
Jijoong Moon
<jijoong.moon@samsung.com>
Tue, 19 Oct 2021 11:34:48 +0000
(20:34 +0900)
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 <jhoon.it.lee@samsung.com>
nntrainer/tensor/manager.cpp
patch
|
blob
|
history
diff --git
a/nntrainer/tensor/manager.cpp
b/nntrainer/tensor/manager.cpp
index ce45a6a22cc02b0c94b6bc4aa535313fac5d1c16..024dd83aefc78f008f98d4e204f3db07047295b3 100644
(file)
--- 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
);
}