#include <OnnxProperty.h>

Inheritance diagram for SourceXtractor::OnnxProperty:

Collaboration diagram for SourceXtractor::OnnxProperty:

[legend]

Classes
struct	NdWrapper
struct	NdWrapperBase

Public Member Functions
virtual	~OnnxProperty ()=default
	OnnxProperty (std::map< std::string, std::unique_ptr< NdWrapperBase > > &&output)
template<typename T>
const Euclid::NdArray::NdArray< T > &	getData (const std::string &key) const
Public Member Functions inherited from SourceXtractor::Property
virtual	~Property ()=default

Private Attributes
std::map< std::string, std::unique_ptr< NdWrapperBase > >	m_output

Detailed Description

Wraps the tensors outputted by the ONNX runtime

Definition at line 30 of file OnnxProperty.h.

Constructor & Destructor Documentation

◆ ~OnnxProperty()

virtual SourceXtractor::OnnxProperty::~OnnxProperty ( )

virtualdefault

Destructor

◆ OnnxProperty()

SourceXtractor::OnnxProperty::OnnxProperty ( std::map< std::string, std::unique_ptr< NdWrapperBase > > && output )

inlineexplicit

Constructor

Parameters

output A map where the key corresponds to the path of one of the ONNX models, and the value to the corresponding output tensor

Definition at line 62 of file OnnxProperty.h.

References m_output, and std::move().

Here is the call graph for this function:

Member Function Documentation

◆ getData()

template<typename T>

const Euclid::NdArray::NdArray< T > & SourceXtractor::OnnxProperty::getData ( const std::string & key ) const

inline

Parameters

key	Path to a registered ONNX model

Returns: Corresponding output tensor

Definition at line 71 of file OnnxProperty.h.

References SourceXtractor::OnnxProperty::NdWrapper< T >::m_ndarray, and m_output.

Member Data Documentation

◆ m_output

std::map<std::string, std::unique_ptr<NdWrapperBase> > SourceXtractor::OnnxProperty::m_output

private

Definition at line 76 of file OnnxProperty.h.

Referenced by getData(), and OnnxProperty().

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

SEImplementation/SEImplementation/Plugin/Onnx/OnnxProperty.h

Classes

Public Member Functions

Private Attributes