concatenate.hpp File Reference

#include "lbann/layers/data_type_layer.hpp"
#include "lbann/utils/dim_helpers.hpp"
#include "lbann/utils/distconv.hpp"
#include "lbann/utils/exception.hpp"
#include "lbann/proto/datatype_helpers.hpp"
#include "lbann/proto/layers.pb.h"
#include <lbann/proto/proto_common.hpp>
#include "lbann/macros/instantiate_device.hpp"

Include dependency graph for concatenate.hpp:

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

Go to the source code of this file.

Classes
class	lbann::concatenate_layer< TensorDataType, Layout, Device >
	Concatenate tensors along specified dimension. More...

Namespaces
	lbann

Macros
#define	PROTO_DEVICE(T, Device)

Functions
template<typename TensorDataType , El::Device Device>
void	lbann::bp_setup_gradient_wrt_inputs_impl (concatenate_layer< TensorDataType, data_layout::MODEL_PARALLEL, Device > &l)
template<typename TensorDataType , El::Device Device>
void	lbann::bp_setup_gradient_wrt_inputs_impl (concatenate_layer< TensorDataType, data_layout::DATA_PARALLEL, Device > &l)
	lbann::PROTO (float)
	lbann::PROTO (double)

Macro Definition Documentation

PROTO_DEVICE

#define PROTO_DEVICE	(		T,
			Device
	)

Value:

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

Definition at line 563 of file concatenate.hpp.