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

#include <OnnxTaskFactory.h>

Inheritance diagram for SourceXtractor::OnnxTaskFactory:
[legend]
Collaboration diagram for SourceXtractor::OnnxTaskFactory:
[legend]

Public Member Functions

 OnnxTaskFactory ()
 
virtual ~OnnxTaskFactory ()=default
 
std::shared_ptr< TaskcreateTask (const PropertyId &property_id) const override
 Returns a Task producing a Property corresponding to the given PropertyId. More...
 
void reportConfigDependencies (Euclid::Configuration::ConfigManager &manager) const override
 Registers all the Configuration dependencies. More...
 
void configure (Euclid::Configuration::ConfigManager &manager) override
 Method which should initialize the object. More...
 
void registerPropertyInstances (OutputRegistry &registry) override
 
- Public Member Functions inherited from SourceXtractor::TaskFactory
virtual ~TaskFactory ()=default
 Destructor. More...
 
- Public Member Functions inherited from SourceXtractor::Configurable
virtual ~Configurable ()=default
 Destructor. More...
 

Private Attributes

std::vector< OnnxModelm_models
 

Detailed Description

Create OnnxSourceTasks

Definition at line 29 of file OnnxTaskFactory.h.

Constructor & Destructor Documentation

SourceXtractor::OnnxTaskFactory::OnnxTaskFactory ( )

Definition at line 65 of file OnnxTaskFactory.cpp.

virtual SourceXtractor::OnnxTaskFactory::~OnnxTaskFactory ( )
virtualdefault

Member Function Documentation

void SourceXtractor::OnnxTaskFactory::configure ( Euclid::Configuration::ConfigManager manager)
overridevirtual

Method which should initialize the object.

The given manager is guaranteed to already be in the initialized state, so all the required Configurations can be accessed using its getConfiguration() method.

Parameters
managerThe manager to get the Configurations from

Reimplemented from SourceXtractor::TaskFactory.

Definition at line 78 of file OnnxTaskFactory.cpp.

References SourceXtractor::formatShape(), SourceXtractor::generatePropertyName(), Euclid::Configuration::ConfigManager::getConfiguration(), SourceXtractor::OnnxConfig::getModels(), Elements::Logging::info(), SourceXtractor::OnnxModel::m_input_name, SourceXtractor::OnnxModel::m_input_shape, SourceXtractor::OnnxModel::m_input_type, SourceXtractor::OnnxModel::m_model_path, m_models, SourceXtractor::OnnxModel::m_output_name, SourceXtractor::OnnxModel::m_output_shape, SourceXtractor::OnnxModel::m_output_type, SourceXtractor::OnnxModel::m_prop_name, SourceXtractor::OnnxModel::m_session, std::move(), SourceXtractor::onnx_logger, and SourceXtractor::ORT_ENV.

Here is the call graph for this function:

std::shared_ptr< Task > SourceXtractor::OnnxTaskFactory::createTask ( const PropertyId property_id) const
overridevirtual

Returns a Task producing a Property corresponding to the given PropertyId.

Implements SourceXtractor::TaskFactory.

Definition at line 67 of file OnnxTaskFactory.cpp.

References m_models.

void SourceXtractor::OnnxTaskFactory::registerPropertyInstances ( OutputRegistry registry)
overridevirtual

Reimplemented from SourceXtractor::TaskFactory.

Definition at line 138 of file OnnxTaskFactory.cpp.

References m_models.

void SourceXtractor::OnnxTaskFactory::reportConfigDependencies ( Euclid::Configuration::ConfigManager manager) const
overridevirtual

Registers all the Configuration dependencies.

Configurable implementations should implement this method to register to the given ConfigManager (using the registerConfiguration() method) all the Configurations they are going to use

Parameters
managerThe manager to register the dependencies to

Reimplemented from SourceXtractor::TaskFactory.

Definition at line 74 of file OnnxTaskFactory.cpp.

References Euclid::Configuration::ConfigManager::registerConfiguration().

Here is the call graph for this function:

Member Data Documentation

std::vector<OnnxModel> SourceXtractor::OnnxTaskFactory::m_models
private

Definition at line 48 of file OnnxTaskFactory.h.

Referenced by configure(), createTask(), and registerPropertyInstances().


The documentation for this class was generated from the following files: