batchwise_reduce_sum.hpp File Reference

#include "lbann/layers/data_type_layer.hpp"
#include "lbann/macros/instantiate_device.hpp"

Include dependency graph for batchwise_reduce_sum.hpp:

Go to the source code of this file.

Classes
class	lbann::batchwise_reduce_sum_layer< TensorDataType, Layout, Device >
	Sum of tensor entries along batch dimension. More...

Namespaces
	lbann

Macros
#define	PROTO_DEVICE(T, Device)

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

Macro Definition Documentation

PROTO_DEVICE

#define PROTO_DEVICE	(		T,
			Device
	)

Value:

extern template class batchwise_reduce_sum_layer<T,                          \
                                                   data_layout::DATA_PARALLEL, \
                                                   Device>;                    \
  extern template class batchwise_reduce_sum_layer<                            \
    T,                                                                         \
    data_layout::MODEL_PARALLEL,                                               \
    Device>;

Definition at line 76 of file batchwise_reduce_sum.hpp.