This is an automated email from the ASF dual-hosted git repository.
wangwei pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/singa.git
The following commit(s) were added to refs/heads/dev by this push:
new 264163b bugfix cpu batchnorm
new 3fcd952 Merge pull request #594 from chrishkchris/hot_fix_dnnl
264163b is described below
commit 264163baf04cc8b55c0ba53329bf0ca22fcde916
Author: chrishkchris <[email protected]>
AuthorDate: Sun Feb 9 07:08:47 2020 +0000
bugfix cpu batchnorm
---
src/model/operation/batchnorm.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/model/operation/batchnorm.cc b/src/model/operation/batchnorm.cc
index 36c5ab1..fa5ce73 100644
--- a/src/model/operation/batchnorm.cc
+++ b/src/model/operation/batchnorm.cc
@@ -177,7 +177,7 @@ const std::vector<Tensor> CpuBatchNormForwardTraining(
{y.block(), running_mean.block(), running_var.block(), mean.block(),
var.block()});
- return {y, running_mean, running_var, mean, var};
+ return {y, mean, var};
}
const std::vector<Tensor> CpuBatchNormBackwardx(
