SourceXtractor::OnnxModel Struct Reference

#include <OnnxModel.h>

Collaboration diagram for SourceXtractor::OnnxModel:

Public Attributes
std::string	m_prop_name
	Name that will be written into the catalog. More...
std::string	m_input_name
	Input tensor name. More...
std::string	m_output_name
	Output tensor name. More...
ONNXTensorElementDataType	m_input_type
	Input type. More...
ONNXTensorElementDataType	m_output_type
	Output type. More...
std::vector< std::int64_t >	m_input_shape
	Input tensor shape. More...
std::vector< std::int64_t >	m_output_shape
	Output tensor shape. More...
std::string	m_model_path
	Path to the ONNX model. More...
std::unique_ptr< Ort::Session >	m_session
	Session, one per model. In theory, it is thread-safe. More...

Detailed Description

Wraps some metadata about a loaded ONNX model, so the Task knows what it needs to pass and receive data from it

Definition at line 32 of file OnnxModel.h.

Member Data Documentation

m_input_name

std::string SourceXtractor::OnnxModel::m_input_name

Input tensor name.

Definition at line 34 of file OnnxModel.h.

Referenced by SourceXtractor::computePropertiesSpecialized(), and SourceXtractor::OnnxTaskFactory::configure().

m_input_shape

std::vector<std::int64_t> SourceXtractor::OnnxModel::m_input_shape

Input tensor shape.

Definition at line 38 of file OnnxModel.h.

Referenced by SourceXtractor::computePropertiesSpecialized(), and SourceXtractor::OnnxTaskFactory::configure().

m_input_type

ONNXTensorElementDataType SourceXtractor::OnnxModel::m_input_type

Input type.

Definition at line 36 of file OnnxModel.h.

Referenced by SourceXtractor::OnnxTaskFactory::configure().

m_model_path

std::string SourceXtractor::OnnxModel::m_model_path

Path to the ONNX model.

Definition at line 40 of file OnnxModel.h.

Referenced by SourceXtractor::OnnxTaskFactory::configure(), and SourceXtractor::registerColumnConverter().

m_output_name

std::string SourceXtractor::OnnxModel::m_output_name

Output tensor name.

Definition at line 35 of file OnnxModel.h.

Referenced by SourceXtractor::computePropertiesSpecialized(), SourceXtractor::OnnxTaskFactory::configure(), and SourceXtractor::generatePropertyName().

m_output_shape

std::vector<std::int64_t> SourceXtractor::OnnxModel::m_output_shape

Output tensor shape.

Definition at line 39 of file OnnxModel.h.

Referenced by SourceXtractor::computePropertiesSpecialized(), and SourceXtractor::OnnxTaskFactory::configure().

m_output_type

ONNXTensorElementDataType SourceXtractor::OnnxModel::m_output_type

Output type.

Definition at line 37 of file OnnxModel.h.

Referenced by SourceXtractor::OnnxTaskFactory::configure().

m_prop_name

std::string SourceXtractor::OnnxModel::m_prop_name

Name that will be written into the catalog.

Definition at line 33 of file OnnxModel.h.

Referenced by SourceXtractor::OnnxTaskFactory::configure(), and SourceXtractor::registerColumnConverter().

m_session

std::unique_ptr<Ort::Session> SourceXtractor::OnnxModel::m_session

Session, one per model. In theory, it is thread-safe.

Definition at line 41 of file OnnxModel.h.

Referenced by SourceXtractor::computePropertiesSpecialized(), SourceXtractor::OnnxTaskFactory::configure(), and SourceXtractor::generatePropertyName().

---

The documentation for this struct was generated from the following file:

• SEImplementation/SEImplementation/Plugin/Onnx/OnnxModel.h