[Fix] Weight initializer stddev calculation
authorJihoon Lee <jhoon.it.lee@samsung.com>
Thu, 1 Jul 2021 05:13:09 +0000 (14:13 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Wed, 7 Jul 2021 03:27:34 +0000 (12:27 +0900)
commit6ab4b7e1c77340843e303d32a9b49495774c1538
treecd15fab5df6ed5f232c1161f97935bd3c6e2df56
parent1e5219bff26dc4623b8418da02d331a78fec7be0
[Fix] Weight initializer stddev calculation

The weight initializer was fitted to only fully connected layer case,
this patch extends weight initializer to properly work with other
weights as well.

**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/weight.cpp