MCPcopy Index your code
hub / github.com/NVIDIA/TensorRT-LLM / get_parameter_tensor

Method get_parameter_tensor

tensorrt_llm/network.py:176–177  ·  view source on GitHub ↗
(self, param)

Source from the content-addressed store, hash-verified

174 return self._parameter_tensors
175
176 def get_parameter_tensor(self, param):
177 return self.parameter_tensors.get(param, None)
178
179 def set_parameter_tensor(self, param, tensor):
180 assert param not in self.parameter_tensors

Callers 3

get_managed_tensorMethod · 0.80
get_constant_tensorMethod · 0.80
_get_weightsMethod · 0.80

Calls 1

getMethod · 0.45

Tested by

no test coverage detected