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 |