MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / to_trt

Method to_trt

tensorrt_llm/python_plugin.py:227–228  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

225 return len(self._dims)
226
227 def to_trt(self) -> trt.DimsExprs:
228 return trt.DimsExprs([i.expr for i in self.dims])
229
230
231class SymTensor:

Callers 1

get_output_shapesMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected