LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
operator_layer_impl.hpp File Reference
#include "lbann/layers/data_type_layer.hpp"
#include "lbann/layers/operator_layer.hpp"
#include "lbann/operators/elementwise_operator.hpp"
#include "lbann/proto/factories.hpp"
#include "lbann/proto/operator_factory.hpp"
#include "lbann/utils/exception.hpp"
#include "lbann/proto/layers.pb.h"
#include <cereal/types/base_class.hpp>
#include <memory>
#include "lbann/macros/instantiate_device.hpp"
Include dependency graph for operator_layer_impl.hpp:

Go to the source code of this file.

Namespaces

 lbann
 

Macros

#define PROTO_DEVICE(T, D)
 

Functions

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

Macro Definition Documentation

◆ PROTO_DEVICE

#define PROTO_DEVICE (   T,
 
)
Value:
extern template class OperatorLayer<T, T, data_layout::DATA_PARALLEL, D>; \
extern template class OperatorLayer<T, T, data_layout::MODEL_PARALLEL, D>; \
extern template std::unique_ptr<Layer> \
build_operator_layer_from_pbuf<T, T, data_layout::DATA_PARALLEL, D>( \
lbann_comm*, \
lbann_data::Layer const&); \
extern template std::unique_ptr<Layer> \
build_operator_layer_from_pbuf<T, T, data_layout::MODEL_PARALLEL, D>( \
lbann_comm*, \
lbann_data::Layer const&)

Definition at line 277 of file operator_layer_impl.hpp.