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

Method get_tensor

tensorrt_llm/parameter.py:144–148  ·  view source on GitHub ↗
(self, network)

Source from the content-addressed store, hash-verified

142 return self._tensor
143
144 def get_tensor(self, network) -> Tensor:
145 if self.is_managed(network):
146 return self.get_managed_tensor(network)
147 else:
148 return self.get_constant_tensor(network)
149
150 def is_managed(self, network):
151 if network is None:

Callers 15

valueMethod · 0.95
dump_statsFunction · 0.80
get_tensorFunction · 0.80
mainFunction · 0.80
convert_and_saveFunction · 0.80
mainFunction · 0.80
get_weightMethod · 0.80
loadFunction · 0.80
kernelMethod · 0.80
kernelMethod · 0.80
kernelMethod · 0.80
kernelMethod · 0.80

Calls 3

is_managedMethod · 0.95
get_managed_tensorMethod · 0.95
get_constant_tensorMethod · 0.95

Tested by

no test coverage detected