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
SEImplementation
Plugin
Onnx
OnnxModel.h
Go to the documentation of this file.
1
19
#ifndef _SEIMPLEMENTATION_PLUGIN_ONNXMODEL_H_
20
#define _SEIMPLEMENTATION_PLUGIN_ONNXMODEL_H_
21
22
#include <string>
23
#include <vector>
24
#include <onnxruntime_cxx_api.h>
25
26
namespace
SourceXtractor {
27
32
struct
OnnxModel
{
33
std::string
m_prop_name
;
34
std::string
m_input_name
;
35
std::string
m_output_name
;
36
ONNXTensorElementDataType
m_input_type
;
37
ONNXTensorElementDataType
m_output_type
;
38
std::vector<std::int64_t>
m_input_shape
;
39
std::vector<std::int64_t>
m_output_shape
;
40
std::string
m_model_path
;
41
std::unique_ptr<Ort::Session>
m_session
;
42
};
43
44
}
// end of namespace SourceXtractor
45
46
#endif // _SEIMPLEMENTATION_PLUGIN_ONNXMODEL_H_
SourceXtractor::OnnxModel
Definition:
OnnxModel.h:32
SourceXtractor::OnnxModel::m_input_type
ONNXTensorElementDataType m_input_type
Input type.
Definition:
OnnxModel.h:36
SourceXtractor::OnnxModel::m_input_shape
std::vector< std::int64_t > m_input_shape
Input tensor shape.
Definition:
OnnxModel.h:38
SourceXtractor::OnnxModel::m_input_name
std::string m_input_name
Input tensor name.
Definition:
OnnxModel.h:34
SourceXtractor::OnnxModel::m_output_name
std::string m_output_name
Output tensor name.
Definition:
OnnxModel.h:35
std::string
STL class.
SourceXtractor::OnnxModel::m_prop_name
std::string m_prop_name
Name that will be written into the catalog.
Definition:
OnnxModel.h:33
SourceXtractor::OnnxModel::m_output_shape
std::vector< std::int64_t > m_output_shape
Output tensor shape.
Definition:
OnnxModel.h:39
std::vector
STL class.
std::unique_ptr< Ort::Session >
SourceXtractor::OnnxModel::m_model_path
std::string m_model_path
Path to the ONNX model.
Definition:
OnnxModel.h:40
SourceXtractor::OnnxModel::m_output_type
ONNXTensorElementDataType m_output_type
Output type.
Definition:
OnnxModel.h:37
SourceXtractor::OnnxModel::m_session
std::unique_ptr< Ort::Session > m_session
Session, one per model. In theory, it is thread-safe.
Definition:
OnnxModel.h:41
Generated by
1.8.5