SourceXtractorPlusPlus
0.15
Please provide a description of the project.
|
#include <OnnxModel.h>
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... | |
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.
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 |
Output tensor name.
Definition at line 35 of file OnnxModel.h.
Referenced by SourceXtractor::computePropertiesSpecialized(), SourceXtractor::OnnxTaskFactory::configure(), and SourceXtractor::generatePropertyName().
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().
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().