OnnxUtils.h File Reference

#include <memory>
#include <vector>
#include <onnxruntime_cxx_api.h>

Include dependency graph for OnnxUtils.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	AthOnnxUtils

Functions
template<typename T >
std::vector< T >	AthOnnxUtils::flattenNestedVectors (const std::vector< std::vector< T >> &features)
void	AthOnnxUtils::getInputNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames)
void	AthOnnxUtils::getOutputNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames)
void	AthOnnxUtils::getNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames, bool isInput)
int64_t	AthOnnxUtils::getTensorSize (const std::vector< int64_t > &dataShape)
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)
template<typename T >
Ort::Value	AthOnnxUtils::createTensor (std::vector< T > &data, const std::vector< int64_t > &dataShape)