SourceXtractorPlusPlus  0.13
Please provide a description of the project.
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 {
36  ONNXTensorElementDataType m_input_type;
37  ONNXTensorElementDataType m_output_type;
42 };
43 
44 } // end of namespace SourceXtractor
45 
46 #endif // _SEIMPLEMENTATION_PLUGIN_ONNXMODEL_H_
std::string
STL class.
SourceXtractor::OnnxModel::m_input_shape
std::vector< std::int64_t > m_input_shape
Input tensor shape.
Definition: OnnxModel.h:38
SourceXtractor::OnnxModel::m_output_name
std::string m_output_name
Output tensor name.
Definition: OnnxModel.h:35
std::vector< std::int64_t >
SourceXtractor
Definition: Aperture.h:30
SourceXtractor::OnnxModel::m_model_path
std::string m_model_path
Path to the ONNX model.
Definition: OnnxModel.h:40
SourceXtractor::OnnxModel::m_input_name
std::string m_input_name
Input tensor name.
Definition: OnnxModel.h:34
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
SourceXtractor::OnnxModel
Definition: OnnxModel.h:32
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
SourceXtractor::OnnxModel::m_output_type
ONNXTensorElementDataType m_output_type
Output type.
Definition: OnnxModel.h:37
std::unique_ptr< Ort::Session >
SourceXtractor::OnnxModel::m_input_type
ONNXTensorElementDataType m_input_type
Input type.
Definition: OnnxModel.h:36