#include "AthOnnxUtils/OnnxUtils.h"
#include <cassert>
#include <string>
Go to the source code of this file.
|
| AthOnnx |
| Namespace holding all of the Onnx Runtime example code.
|
|
|
void | AthOnnx::getNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames, bool isInput) |
|
void | AthOnnx::getInputNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames) |
|
void | AthOnnx::getOutputNodeInfo (const Ort::Session &session, std::vector< std::vector< int64_t > > &dataShape, std::vector< std::string > &nodeNames) |
|
void | AthOnnx::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) |
|
int64_t | AthOnnx::getTensorSize (const std::vector< int64_t > &dataShape) |
|
Ort::Value | AthOnnx::createTensor (std::vector< float > &data, const std::vector< int64_t > &dataShape) |
|