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

Method get_managed_tensor

tensorrt_llm/parameter.py:94–101  ·  view source on GitHub ↗
(self, network: Network)

Source from the content-addressed store, hash-verified

92 return Tensor(name=self._name, dtype=self._dtype, shape=self._shape)
93
94 def get_managed_tensor(self, network: Network) -> Tensor:
95 if self._network is None or self._network() != network:
96 self._network = weakref.ref(network)
97 self._tensor = network.get_parameter_tensor(self)
98 if self._tensor is None:
99 self._tensor = self._create_managed_tensor(network)
100 network.set_parameter_tensor(self, self._tensor)
101 return self._tensor
102
103 def _create_constant_tensor(self) -> Tensor:
104 if (self._value is not None and isinstance(self._value, np.ndarray)

Callers 2

get_tensorMethod · 0.95
get_weightMethod · 0.80

Calls 3

get_parameter_tensorMethod · 0.80
set_parameter_tensorMethod · 0.80

Tested by

no test coverage detected