Method from_trt_desc

tensorrt_llm/_utils.py:1050–1051 · view source on GitHub ↗

(desc: trt.PluginTensorDesc, pointer: int)

Source from the content-addressed store, hash-verified

1048
1049	@staticmethod
1050	def from_trt_desc(desc: trt.PluginTensorDesc, pointer: int):
1051	return TensorWrapper(pointer, trt_dtype_to_torch(desc.type), desc.dims)
1052
1053
1054	def convert_to_torch_tensor(

enqueueMethod · 0.80

TensorWrapperClass · 0.85

trt_dtype_to_torchFunction · 0.70

no test coverage detected