#include <OnnxProperty.h>

Inheritance diagram for SourceXtractor::OnnxProperty:

Collaboration diagram for SourceXtractor::OnnxProperty:

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 )

inline

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 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 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().

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

SEImplementation/SEImplementation/Plugin/Onnx/OnnxProperty.h

Classes

Public Member Functions

Private Attributes