AthOnnxUtils Namespace Reference

Functions
template<typename T>
std::vector< T >	flattenNestedVectors (const std::vector< std::vector< T > > &features)
void	getInputNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames)
void	getOutputNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames)
void	getNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames, bool isInput)
int64_t	getTensorSize (const std::vector< int64_t > &dataShape)
void	inferenceWithIOBinding (Ort::Session &session, const std::vector< std::string > &inputNames, const std::vector< Ort::Value > &inputData, const std::vector< std::string > &outputNames, const std::vector< Ort::Value > &outputData)
template<typename T>
Ort::Value	createTensor (std::vector< T > &data, const std::vector< int64_t > &dataShape)

Function Documentation

createTensor()

template<typename T>

Ort::Value AthOnnxUtils::createTensor	(	std::vector< T > &	data,
		const std::vector< int64_t > &	dataShape )

Definition at line 78 of file OnnxUtils.h.

flattenNestedVectors()

template<typename T>

std::vector< T > AthOnnxUtils::flattenNestedVectors ( const std::vector< std::vector< T > > & features )

inline

Definition at line 20 of file OnnxUtils.h.

getInputNodeInfo()

void AthOnnxUtils::getInputNodeInfo	(	const Ort::Session &	session,
		std::vector< std::vector< int64_t > > &	dataShape,
		std::vector< std::string > &	nodeNames )

Definition at line 33 of file OnnxUtils.cxx.

getNodeInfo()

void AthOnnxUtils::getNodeInfo	(	const Ort::Session &	session,
		std::vector< std::vector< int64_t > > &	dataShape,
		std::vector< std::string > &	nodeNames,
		bool	isInput )

Definition at line 9 of file OnnxUtils.cxx.

getOutputNodeInfo()

void AthOnnxUtils::getOutputNodeInfo	(	const Ort::Session &	session,
		std::vector< std::vector< int64_t > > &	dataShape,
		std::vector< std::string > &	nodeNames )

Definition at line 41 of file OnnxUtils.cxx.

getTensorSize()

int64_t AthOnnxUtils::getTensorSize

(

const std::vector< int64_t > &

dataShape

)

Definition at line 73 of file OnnxUtils.cxx.

inferenceWithIOBinding()

void AthOnnxUtils::inferenceWithIOBinding	(	Ort::Session &	session,
		const std::vector< std::string > &	inputNames,
		const std::vector< Ort::Value > &	inputData,
		const std::vector< std::string > &	outputNames,
		const std::vector< Ort::Value > &	outputData )

Definition at line 49 of file OnnxUtils.cxx.