onnx-mlir

Logo

ONNX 模型在 MLIR 编译器基础设施中的表示和参考下推

在 GitHub 上查看项目 onnx/onnx-mlir

操作指南

使用 Python 进行推理
使用 C/C++ 进行推理
使用 Java 进行推理

参考资料

ONNX 方言
OMTensor C99 运行时 API
OMTensorList C99 运行时 API
OMTensor Java 运行时 API
OMTensorList Java 运行时 API
生成 ONNX 方言
关于文档

开发

添加操作
测试指南
错误处理
命令行选项
插桩
常量传播
添加加速器

工具

工具

RunONNXModel.py
DocCheck

此项目由 onnx 维护

托管于 GitHub Pages — 主题来自 orderedlist

onnx-mlir: onnx-mlir/include/OnnxMlirRuntime.h 源代码文件 - ONNX 开放神经网络交换
onnx-mlir
加载中...
搜索中...
无匹配项
OnnxMlirRuntime.h
转到此文件的文档。
1/*
2 * SPDX-License-Identifier: Apache-2.0
3 */
4
5//===------- OnnxMlirRuntime.h - ONNX-MLIR 运行时 API 声明 -------===//
6//
7// Copyright 2019-2023 The IBM Research Authors.
8//
9// =============================================================================
10//
11// 此文件包含外部 OMTensor 数据结构和
12// 辅助函数的声明。
13//
14//===----------------------------------------------------------------------===//
15#ifndef ONNX_MLIR_ONNXMLIRRUNTIME_H
16#define ONNX_MLIR_ONNXMLIRRUNTIME_H
17
18#ifdef __cplusplus
19#include <cstdint>
20#else
21#include <stdbool.h>
22#include <stdint.h>
23#endif
24
25#include <onnx-mlir/Runtime/OMEntryPoint.h>
26#include <onnx-mlir/Runtime/OMInstrument.h>
27#include <onnx-mlir/Runtime/OMSignature.h>
28#include <onnx-mlir/Runtime/OMTensor.h>
29#include <onnx-mlir/Runtime/OMTensorList.h>
30
180#endif // ONNX_MLIR_ONNXMLIRRUNTIME_H