batch_normalization.hpp File Reference

#include "lbann/layers/data_type_layer.hpp"
#include "lbann/models/model.hpp"
#include "lbann/proto/datatype_helpers.hpp"
#include "lbann/proto/layers.pb.h"
#include "lbann/macros/instantiate_device.hpp"

Include dependency graph for batch_normalization.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	lbann::batch_normalization_layer< TensorDataType, T_layout, Dev >
	Channel-wise batch normalization, including scale/bias. More...

Namespaces
	lbann

Macros
#define	PROTO_DEVICE(T, Device)

Enumerations
enum	lbann::batch_normalization_stats_aggregation { lbann::batch_normalization_stats_aggregation::local, lbann::batch_normalization_stats_aggregation::node_local, lbann::batch_normalization_stats_aggregation::global }

Functions
	lbann::LBANN_DEFINE_LAYER_BUILDER (batch_normalization)
	lbann::PROTO (float)
	lbann::PROTO (double)

Macro Definition Documentation

PROTO_DEVICE

#define PROTO_DEVICE	(		T,
			Device
	)

Value:

extern template class batch_normalization_layer<T,                           \
                                                  data_layout::DATA_PARALLEL,  \
                                                  Device>

Definition at line 478 of file batch_normalization.hpp.