SourceXtractor::OnnxPlugin Class Reference

#include <OnnxPlugin.h>

Inheritance diagram for SourceXtractor::OnnxPlugin:

Collaboration diagram for SourceXtractor::OnnxPlugin:

Public Member Functions
virtual	~OnnxPlugin ()=default
	Destructor.
std::string	getIdString () const override
void	registerPlugin (PluginAPI &plugin_api) override
Public Member Functions inherited from SourceXtractor::Plugin
virtual	~Plugin ()=default

Detailed Description

The ONNX plugin wraps a machine learning model stored using the "Open Neural Network Exchange" format. SourceXtractor will feed the model a stamp with the size the model requires as an input. For the output column, the 'domain' field + the graph name from the ONNX model will be used: i.e. ch.unige.astro_galaxy_class

See also

https://onnx.ai/

Definition at line 35 of file OnnxPlugin.h.

Constructor & Destructor Documentation

~OnnxPlugin()

virtual SourceXtractor::OnnxPlugin::~OnnxPlugin ( )

virtualdefault

Destructor.

Member Function Documentation

getIdString()

std::string SourceXtractor::OnnxPlugin::getIdString ( ) const

overridevirtual

Implements SourceXtractor::Plugin.

Definition at line 28 of file OnnxPlugin.cpp.

registerPlugin()

void SourceXtractor::OnnxPlugin::registerPlugin ( PluginAPI & plugin_api )

overridevirtual

Implements SourceXtractor::Plugin.

Definition at line 32 of file OnnxPlugin.cpp.

References SourceXtractor::OutputRegistry::enableOutput(), SourceXtractor::PluginAPI::getOutputRegistry(), SourceXtractor::PluginAPI::getTaskFactoryRegistry(), SourceXtractor::OutputRegistry::registerColumnConverter(), and SourceXtractor::TaskFactoryRegistry::registerTaskFactory().

Here is the call graph for this function:

---

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

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