SourceXtractorPlusPlus
0.12
Please provide a description of the project.
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
Definition:
Aperture.h:30
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< std::int64_t >
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.14