polygraphy.backend.trt.loader.EngineFromBytes Class Reference

Inheritance diagram for polygraphy.backend.trt.loader.EngineFromBytes:

Collaboration diagram for polygraphy.backend.trt.loader.EngineFromBytes:

Public Member Functions
def	__init__ (self, serialized_engine)
def	__call__ (self)

Private Attributes
	_serialized_engine

Constructor & Destructor Documentation

__init__()

def polygraphy.backend.trt.loader.EngineFromBytes.__init__	(		self,
			serialized_engine
	)

Functor that deserializes an engine from a buffer.

Args:
    serialized_engine (Callable() -> Union[str, bytes]):
    Either a loader that can supply a memory buffer, or a memory buffer itself.
    For example, use a lambda: `lambda: open("/path/to/trt.engine", "rb").read()` for lazy evaluation,
    or the contents of the file directly: `open("/path/to/trt.engine", "rb").read()` for immediate evaluation.

Member Function Documentation

__call__()

def polygraphy.backend.trt.loader.EngineFromBytes.__call__

(

self

)

Deserializes an engine from a buffer.

Returns:
    trt.ICudaEngine: The deserialized engine.

Member Data Documentation

_serialized_engine

polygraphy.backend.trt.loader.EngineFromBytes._serialized_engine

private

---

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

• backend/trt/loader.py