SourceXtractor::OnnxTaskFactory Class Reference

#include <OnnxTaskFactory.h>

Inheritance diagram for SourceXtractor::OnnxTaskFactory:

Collaboration diagram for SourceXtractor::OnnxTaskFactory:

Public Member Functions
	OnnxTaskFactory ()
virtual	~OnnxTaskFactory ()=default
std::shared_ptr< Task >	createTask (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< OnnxModel >	m_models

Detailed Description

Create OnnxSourceTasks

Definition at line 29 of file OnnxTaskFactory.h.

Constructor & Destructor Documentation

OnnxTaskFactory()

SourceXtractor::OnnxTaskFactory::OnnxTaskFactory

(

)

Definition at line 65 of file OnnxTaskFactory.cpp.

~OnnxTaskFactory()

virtual SourceXtractor::OnnxTaskFactory::~OnnxTaskFactory ( )

virtualdefault

Member Function Documentation

configure()

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

manager

The 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, SourceXtractor::ORT_ENV, and std::vector< T >::size().

Here is the call graph for this function:

createTask()

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.

registerPropertyInstances()

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

overridevirtual

Reimplemented from SourceXtractor::TaskFactory.

Definition at line 138 of file OnnxTaskFactory.cpp.

References m_models.

reportConfigDependencies()

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

manager

The 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

m_models

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:

• SEImplementation/SEImplementation/Plugin/Onnx/OnnxTaskFactory.h
• SEImplementation/src/lib/Plugin/Onnx/OnnxTaskFactory.cpp