[singa] branch dev updated: bugfix cpu batchnorm

[wa...@apache.org]

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 <ch...@yahoo.com.hk>
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(