[
https://issues.apache.org/jira/browse/SINGA-267?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15599750#comment-15599750
]
ASF subversion and git services commented on SINGA-267:
-------------------------------------------------------
Commit fac3af94990e4c9977f82a11ae85f89de9dbb463 in incubator-singa's branch
refs/heads/master from WANG Ji
[ https://git-wip-us.apache.org/repos/asf?p=incubator-singa.git;h=fac3af9 ]
SINGA-267 Add spatial mode in batch normalization layer
Revise batchnorm test cases for spatial mode.
> Add spatial mode in batch normalization layer
> ----------------------------------------------
>
> Key: SINGA-267
> URL: https://issues.apache.org/jira/browse/SINGA-267
> Project: Singa
> Issue Type: Improvement
> Reporter: Wang Ji
>
> Add spatial mode in batch normalization layer in C++ implementation, which
> corresponds to CUDNN_BATCHNORM_SPATIAL in CuDNN.
> Also add logics to automatically detect proper modes in batch normalization
> layer, i.e., if input is 2D tensor then batchnorm layer chooses
> PER_ACTIVATION mode, if input is 4D tensor then batchnorm layer chooses
> SPATIAL mode.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
