11 #ifndef EXATN_NUMERICS_NETWORK_BUILD_FACTORY_HPP_
12 #define EXATN_NUMERICS_NETWORK_BUILD_FACTORY_HPP_
14 #include "tensor_basic.hpp"
15 #include "network_builder.hpp"
16 #include "network_builder_mps.hpp"
17 #include "network_builder_tree.hpp"
51 std::map<std::string,createNetworkBuilderFn> factory_map_;
59 #endif //EXATN_NUMERICS_NETWORK_BUILD_FACTORY_HPP_
Definition: DriverClient.hpp:10
static NetworkBuildFactory * get()
Definition: network_build_factory.cpp:37
void registerNetworkBuilder(const std::string &name, createNetworkBuilderFn creator)
Definition: network_build_factory.cpp:19
std::unique_ptr< NetworkBuilder > createNetworkBuilder(const std::string &name)
Definition: network_build_factory.cpp:25
Definition: network_build_factory.hpp:27
std::shared_ptr< NetworkBuilder > createNetworkBuilderShared(const std::string &name)
Definition: network_build_factory.cpp:32