◆ __init__()
def polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes.__init__ |
( |
|
self, |
|
|
|
model_bytes |
|
) |
| |
Functor that builds an ONNX-Runtime inference session.
Args:
model_bytes (Callable() -> bytes): A loader that can supply a serialized ONNX model.
◆ __call__()
def polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes.__call__ |
( |
|
self | ) |
|
Builds an ONNX-Runtime inference session.
Returns:
onnxruntime.InferenceSession: The inference session.
◆ _model_bytes
polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes._model_bytes |
|
private |
The documentation for this class was generated from the following file: