TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes Class Reference
Inheritance diagram for polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes:
Collaboration diagram for polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes:

Public Member Functions

def __init__ (self, model_bytes)
 
def __call__ (self)
 

Private Attributes

 _model_bytes
 

Constructor & Destructor Documentation

◆ __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.

Member Function Documentation

◆ __call__()

def polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes.__call__ (   self)
Builds an ONNX-Runtime inference session.

Returns:
    onnxruntime.InferenceSession: The inference session.

Member Data Documentation

◆ _model_bytes

polygraphy.backend.onnxrt.loader.SessionFromOnnxBytes._model_bytes
private

The documentation for this class was generated from the following file: