stop_gradient.hpp File Reference

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

Include dependency graph for stop_gradient.hpp:

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

Go to the source code of this file.

Classes
class	lbann::stop_gradient_layer< TensorDataType, T_layout, Dev >
	Block error signals during back propagation. 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 stop_gradient_layer<T,                                 \
                                            data_layout::DATA_PARALLEL,        \
                                            Device>;                           \
  extern template class stop_gradient_layer<T,                                 \
                                            data_layout::MODEL_PARALLEL,       \
                                            Device>

Definition at line 87 of file stop_gradient.hpp.