LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
factories.hpp File Reference
#include "lbann/data_readers/data_reader.hpp"
#include "lbann/layers/layer.hpp"
#include "lbann/operators/operator.hpp"
#include "lbann/proto/proto_common.hpp"
#include "lbann/transforms/transform.hpp"
#include "lbann/transforms/transform_pipeline.hpp"
#include <google/protobuf/message.h>
#include <map>
#include <memory>
Include dependency graph for factories.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 lbann_data
 
 lbann
 
 lbann::proto
 

Functions

std::unique_ptr< trainer > lbann::proto::construct_trainer (lbann_comm *comm, const lbann_data::Trainer &proto_trainer)
 
std::unique_ptr< model > lbann::proto::construct_model (lbann_comm *comm, const lbann_data::Optimizer &proto_opt, const lbann_data::Trainer &proto_trainer, const lbann_data::Model &proto_model)
 
std::vector< OwningLayerPtr > lbann::proto::construct_layer_graph (lbann_comm *comm, const lbann_data::Trainer &proto_trainer, const lbann_data::Model &proto_model)
 
template<typename TensorDataType , data_layout layout, El::Device Dev>
std::unique_ptr< Layer > lbann::proto::construct_layer (lbann_comm *comm, const lbann_data::Layer &proto_layer)
 
template<typename InputT , typename OutputT , El::Device D>
auto lbann::proto::construct_operator (const lbann_data::Operator &proto_operator) -> std::unique_ptr< Operator< InputT, OutputT, D >>
 
std::unique_ptr< weights > lbann::proto::construct_weights (lbann_comm *comm, const lbann_data::Optimizer &proto_opt, const lbann_data::Weights &proto_weights)
 
std::unique_ptr< callback_base > lbann::proto::construct_callback (const google::protobuf::Message &proto_cb)
 
std::unique_ptr< callback_base > lbann::proto::construct_callback (const google::protobuf::Message &proto_cb, std::shared_ptr< lbann_summary > const &summarizer)
 
std::unique_ptr< lbann_summary > lbann::proto::construct_summarizer (lbann_comm *comm, const lbann_data::Model &m)
 
template<typename T >
std::unique_ptr< optimizer > lbann::proto::construct_optimizer (const lbann_data::Optimizer &proto_opt)
 
std::unique_ptr< objective_function > lbann::proto::construct_objective_function (const lbann_data::ObjectiveFunction &proto_obj)
 
std::unique_ptr< transform::transform > lbann::proto::construct_transform (const lbann_data::Transform &trans)
 
transform::transform_pipeline lbann::proto::construct_transform_pipeline (const lbann_data::Reader &data_reader)