ONNXRuntimeAdapter#
- class ONNXRuntimeAdapter[source]#
Bases:
object
Wrapper around ONNX-runtime that behaves like a PyTorch nn.Module.
That is, it implements __call__() and accepts PyTorch Tensors as inputs and also outputs PyTorch Tensors.
- __init__(ort_session: InferenceSession) None [source]#
Constructor.
- Parameters
ort_session (ort.InferenceSession) – ONNX-runtime InferenceSession.
- Return type
None
Methods
- __init__(ort_session: InferenceSession) None [source]#
Constructor.
- Parameters
ort_session (ort.InferenceSession) – ONNX-runtime InferenceSession.
- Return type
None