You are viewing a plain text version of this content. The canonical link for it is here.

Posted to dev@singa.apache.org by "Wang Ji (JIRA)" <ji...@apache.org> on 2016/10/21 14:09:58 UTC

[jira] [Created] (SINGA-267) Add spatial mode in batch normalization layer

Wang Ji created SINGA-267:
-----------------------------

             Summary: 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)