Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
View the Project on GitHub onnx/onnx-mlir
This project is maintained by onnx
Hosted on GitHub Pages — Theme by orderedlist
|
onnx-mlir
|
Public Member Functions | |
| OMTensorList (OMTensor[] omts) | |
| OMTensor[] | getOmtArray () |
| OMTensor | getOmtByIndex (int index) |
Class holding an array of OMTensor so that they can be passed into and out of the compiled model runtime entry point as inputs and outputs.
|
inline |
Constructor
| omts | OMTensor array |
|
inline |
OMTensor array getter
|
inline |
OMTensor getter
| index | index of OMTensor to get |