SourceXtractorPlusPlus  0.13
Please provide a description of the project.
Classes | Public Member Functions | Private Attributes | List of all members
SourceXtractor::OnnxProperty Class Reference

#include <OnnxProperty.h>

Inheritance diagram for SourceXtractor::OnnxProperty:
[legend]
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)
inline

Constructor

Parameters
outputA 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
keyPath 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: