polygraphy.backend.trt.loader.CreateNetwork Class Reference

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

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

Public Member Functions
def	__init__ (self, explicit_precision=None, explicit_batch=None)
def	__call__ (self)

Public Attributes
	explicit_precision
	explicit_batch

Constructor & Destructor Documentation

__init__()

def polygraphy.backend.trt.loader.CreateNetwork.__init__	(		self,
			explicit_precision = None,
			explicit_batch = None
	)

Functor that creates an empty TensorRT network.

Args:
    explicit_precision (bool):
    Whether to create the network with explicit precision enabled. Defaults to False
    explicit_batch (bool):
    Whether to create the network with explicit batch mode. Defaults to True.

Member Function Documentation

__call__()

def polygraphy.backend.trt.loader.CreateNetwork.__call__

(

self

)

Creates an empty TensorRT network.

Returns:
    (trt.Builder, trt.INetworkDefinition): The builder and empty network.

Member Data Documentation

explicit_precision

polygraphy.backend.trt.loader.CreateNetwork.explicit_precision

explicit_batch

polygraphy.backend.trt.loader.CreateNetwork.explicit_batch

---

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

• backend/trt/loader.py