SourceXtractorPlusPlus  0.15
Please provide a description of the project.
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Public Attributes | List of all members
SourceXtractor::OnnxModel Struct Reference

#include <OnnxModel.h>

Collaboration diagram for SourceXtractor::OnnxModel:
[legend]

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_tm_input_shape
 Input tensor shape. More...
 
std::vector< std::int64_tm_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

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().

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().

ONNXTensorElementDataType SourceXtractor::OnnxModel::m_input_type

Input type.

Definition at line 36 of file OnnxModel.h.

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

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().

std::string SourceXtractor::OnnxModel::m_output_name
std::vector<std::int64_t> SourceXtractor::OnnxModel::m_output_shape
ONNXTensorElementDataType SourceXtractor::OnnxModel::m_output_type

Output type.

Definition at line 37 of file OnnxModel.h.

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

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().

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: