OnnxUtils.cxx File Reference

#include "AthOnnxUtils/OnnxUtils.h"
#include <cassert>
#include <string>
#include "AthenaBaseComps/AthAsynchronousAlgorithm.h"
#include <boost/fiber/all.hpp>

Include dependency graph for OnnxUtils.cxx:

Go to the source code of this file.

Namespaces
namespace	AthOnnxUtils

Functions
void	AthOnnxUtils::getNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames, bool isInput)
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::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)
std::string	AthOnnxUtils::asyncInference (Ort::Session &session, const std::vector< std::string > &inputNames, const std::vector< Ort::Value > &inputData, const std::vector< std::string > &outputNames, std::vector< Ort::Value > &outputData, const AthAsynchronousAlgorithm *parentAlg)
int64_t	AthOnnxUtils::getTensorSize (const std::vector< int64_t > &dataShape)