SourceXtractorPlusPlus
0.14
Please provide a description of the project.
Main Page
Related Pages
Namespaces
Classes
Files
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Groups
Pages
SEImplementation
src
lib
Plugin
Onnx
OnnxConfig.cpp
Go to the documentation of this file.
1
18
#include "
SEImplementation/Plugin/Onnx/OnnxConfig.h
"
19
#include <boost/program_options.hpp>
20
21
namespace
po = boost::program_options;
22
using namespace
Euclid::Configuration;
23
24
namespace
SourceXtractor {
25
26
static
const
std::string
ONNX_MODEL
{
"onnx-model"
};
27
28
OnnxConfig::OnnxConfig(
long
manager_id) :
Configuration
(manager_id) {
29
}
30
31
auto
OnnxConfig::getProgramOptions
() ->
std::map<std::string, OptionDescriptionList>
{
32
return
{{
"ONNX"
, {
33
{
ONNX_MODEL
.
c_str
(), po::value<std::vector<std::string>>()->multitoken(),
"ONNX models"
}
34
}}};
35
}
36
37
void
OnnxConfig::initialize
(
const
Configuration::Configuration::UserValues& args) {
38
auto
i = args.find(
ONNX_MODEL
);
39
if
(i != args.end()) {
40
m_onnx_model_paths
= i->second.as<
std::vector<std::string>
>();
41
}
42
}
43
44
}
// end of namespace SourceXtractor
Euclid::Configuration::Configuration
SourceXtractor::OnnxConfig::initialize
void initialize(const UserValues &args) override
Definition:
OnnxConfig.cpp:37
SourceXtractor::ONNX_MODEL
static const std::string ONNX_MODEL
Definition:
OnnxConfig.cpp:26
std::map
STL class.
SourceXtractor::OnnxConfig::m_onnx_model_paths
std::vector< std::string > m_onnx_model_paths
Definition:
OnnxConfig.h:49
std::string
STL class.
SourceXtractor::OnnxConfig::getProgramOptions
std::map< std::string, OptionDescriptionList > getProgramOptions() override
Definition:
OnnxConfig.cpp:31
std::vector< std::string >
std::string::c_str
T c_str(T...args)
OnnxConfig.h
Generated by
1.8.5