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)
