SourceXtractorPlusPlus  0.14
Please provide a description of the project.
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
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_
ONNXTensorElementDataType m_input_type
Input type.
Definition: OnnxModel.h:36
std::vector< std::int64_t > m_input_shape
Input tensor shape.
Definition: OnnxModel.h:38
std::string m_input_name
Input tensor name.
Definition: OnnxModel.h:34
std::string m_output_name
Output tensor name.
Definition: OnnxModel.h:35
STL class.
std::string m_prop_name
Name that will be written into the catalog.
Definition: OnnxModel.h:33
std::vector< std::int64_t > m_output_shape
Output tensor shape.
Definition: OnnxModel.h:39
STL class.
std::string m_model_path
Path to the ONNX model.
Definition: OnnxModel.h:40
ONNXTensorElementDataType m_output_type
Output type.
Definition: OnnxModel.h:37
std::unique_ptr< Ort::Session > m_session
Session, one per model. In theory, it is thread-safe.
Definition: OnnxModel.h:41