MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / __init__

Method __init__

demo/HuggingFace/NNDF/tensorrt_utils.py:297–302  ·  view source on GitHub ↗
(self, engine_fpath: str, network_metadata: NetworkMetadata)

Source from the content-addressed store, hash-verified

295 """
296
297 def __init__(self, engine_fpath: str, network_metadata: NetworkMetadata):
298 self.network_metadata = network_metadata
299
300 self.trt_engine = engine_from_bytes(bytes_from_path(engine_fpath))
301 self.trt_context = TrtRunner(self.trt_engine.create_execution_context())
302 self.trt_context.activate()
303
304 def __call__(self, *args, **kwargs):
305 # hook polygraphy verbosity for inference

Callers

nothing calls this directly

Calls 2

TrtRunnerClass · 0.90
activateMethod · 0.45

Tested by

no test coverage detected