operator_factory_impl.hpp File Reference

#include "lbann_config.hpp"
#include "lbann/proto/factories.hpp"
#include "lbann/proto/operator_factory.hpp"
#include "lbann/operators/activations/activation_builders.hpp"
#include "lbann/operators/loss/loss_builders.hpp"
#include "lbann/operators/math/math_builders.hpp"
#include "lbann/operators/operator.hpp"
#include "lbann/proto/datatype_helpers.hpp"
#include "lbann/utils/protobuf.hpp"
#include <lbann/macros/instantiate_device.hpp>

Include dependency graph for operator_factory_impl.hpp:

Go to the source code of this file.

Namespaces
	lbann
	lbann::proto
	lbann::proto::details

Macros
#define	LBANN_REGISTER_BUILDER(OP_NAME, OP_LOWER)
#define	PROTO_DEVICE(T, DEVICE)

Functions
template<typename InT , typename OutT , El::Device D>
OperatorFactory< InT, OutT, D >	lbann::proto::details::build_default_factory ()
	lbann::PROTO (float)
	lbann::PROTO (double)

Macro Definition Documentation

LBANN_REGISTER_BUILDER

#define LBANN_REGISTER_BUILDER	(		OP_NAME,
			OP_LOWER
	)

Value:

factory.register_builder(#OP_NAME "Operator",                                \
                           build_##OP_LOWER##_operator<InT, D>)

PROTO_DEVICE

#define PROTO_DEVICE	(		T,
			DEVICE
	)

Value:

extern template proto::OperatorFactory<T, T, DEVICE>&                        \
  proto::get_operator_factory();                                               \
  extern template std::unique_ptr<Operator<T, T, DEVICE>>                      \
  proto::construct_operator(lbann_data::Operator const& msg)

Definition at line 164 of file operator_factory_impl.hpp.