Supported Layers - naoyam/lbann GitHub Wiki

Conditions are checked at member function Layer::using_distconv. See the function for full details.

Convolution

• Padding must match the filter dimension, i.e., (filter_dimension - 1) / 2 == padding.

Pooling

• Only max pooling is supported.
• Padding is not supported yet.

Batch normalization

• Global statistics are always used when decomposed in dimensions other than samples.

ReLU