onnx.h

Go to the documentation of this file.

//=== MLModelRunner/ONNXModelRunner/onnx.h --- ONNX C++ Interface -----===//
//
// Part of the MLCompilerBridge Project, under the Apache License v2.0 with LLVM
// Exceptions. See the LICENSE file for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===---------------------------------------------------------------------===//
//===---------------------------------------------------------------------===//
 
#ifndef ONNX_MODELRUNNER_ONNX_H
#define ONNX_MODELRUNNER_ONNX_H
 
#include "llvm/ADT/SmallVector.h"
 
#include <cstdint>
#include <vector>
 
namespace Ort {
class Env;
class Session;
class Value;
} // namespace Ort
 
class ONNXModel {
  Ort::Env *env;
  const char *model_path;
  Ort::Session *session;
  Ort::Value getInputValue(llvm::SmallVector<float, 100> &input, int inputIdx);
 
public:
  ONNXModel(const char *model_path);
 
  void run(llvm::SmallVector<float, 100> &input,
           llvm::SmallVector<float, 100> &output);
};
 
#endif // ONNX_MODELRUNNER_ONNX_H