polygraphy.backend.trt.loader.SaveEngine Class Reference

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

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

Public Member Functions
def	__init__ (self, engine, path=None)
def	__call__ (self)

Public Attributes
	path

Private Attributes
	_engine

Constructor & Destructor Documentation

__init__()

def polygraphy.backend.trt.loader.SaveEngine.__init__	(		self,
			engine,
			path = None
	)

Functor that saves an engine to the provided path.

Args:
    engine (Callable() -> trt.ICudaEngine):
    A callable that can supply a TensorRT engine.


    path (str): The path at which to save the engine.

Member Function Documentation

__call__()

def polygraphy.backend.trt.loader.SaveEngine.__call__

(

self

)

Saves an engine to the provided path.

Returns:
    trt.ICudaEngine: The engine that was saved.

Member Data Documentation

_engine

polygraphy.backend.trt.loader.SaveEngine._engine

private

path

polygraphy.backend.trt.loader.SaveEngine.path

---

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

• backend/trt/loader.py