math_builders_impl.hpp File Reference

#include "lbann/operators/math/math_builders.hpp"
#include "lbann/operators/math/abs.hpp"
#include "lbann/operators/math/binary.hpp"
#include "lbann/operators/math/binary_with_constant.hpp"
#include "lbann/operators/math/clamp.hpp"
#include "lbann/operators/math/select.hpp"
#include "lbann/operators/math/unary.hpp"
#include "lbann/proto/datatype_helpers.hpp"
#include "lbann/proto/operators.pb.h"

Include dependency graph for math_builders_impl.hpp:

Go to the source code of this file.

Macros
#define	LBANN_DEFINE_BIN_WITH_CONSTANT_BUILDER(OP_NAME, OP_LOWER_NAME)

Macro Definition Documentation

LBANN_DEFINE_BIN_WITH_CONSTANT_BUILDER

#define LBANN_DEFINE_BIN_WITH_CONSTANT_BUILDER	(		OP_NAME,
			OP_LOWER_NAME
	)

Value:

template <typename DataT, El::Device D>                                      \
  std::unique_ptr<lbann::Operator<DataT, DataT, D>>                            \
    lbann::build_##OP_LOWER_NAME##_operator(lbann_data::Operator const& op)    \
  {                                                                            \
    details::AssertConsistentTypeParameters<DataT, DataT, D>(op);              \
    lbann_data::OP_NAME##Operator params;                                      \
    LBANN_ASSERT(op.parameters().UnpackTo(&params));                           \
    return std::make_unique<OP_NAME##Operator<DataT, D>>(params.constant());   \
  }

Definition at line 74 of file math_builders_impl.hpp.