layers/transform/pooling.hpp File Reference

#include "lbann/layers/data_type_layer.hpp"
#include "lbann/models/model.hpp"
#include "lbann/utils/dim_helpers.hpp"
#include "lbann/utils/dnn_enums.hpp"
#include "lbann/utils/exception.hpp"
#include "lbann/utils/im2col.hpp"
#include <utility>
#include <vector>
#include "lbann/macros/instantiate_device.hpp"

Include dependency graph for layers/transform/pooling.hpp:

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

Go to the source code of this file.

Classes
class	lbann::unpooling_layer< TensorDataType, T_layout, Dev >
	Transpose of pooling layer. More...
class	lbann::pooling_layer< TensorDataType, T_layout, Dev >

Namespaces
	lbann

Macros
#define	PROTO_DEVICE(T, Device)   extern template class pooling_layer<T, data_layout::DATA_PARALLEL, Device>

Functions
pooling_mode	lbann::to_pool_mode (std::string m)
	lbann::PROTO (float)
	lbann::PROTO (double)

Macro Definition Documentation

PROTO_DEVICE

#define PROTO_DEVICE	(		T,
			Device
	)		extern template class pooling_layer<T, data_layout::DATA_PARALLEL, Device>

Definition at line 435 of file layers/transform/pooling.hpp.