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

Method get_constant_tensor

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

Source from the content-addressed store, hash-verified

132 return tensor
133
134 def get_constant_tensor(self, network: Network) -> Tensor:
135 if self._network is None or self._network() != network:
136 self._network = weakref.ref(network)
137 self._tensor = network.get_parameter_tensor(self)
138 if self._tensor is None:
139 self._tensor = self._create_constant_tensor()
140 self._name = self._tensor.producer.name
141 network.set_parameter_tensor(self, self._tensor)
142 return self._tensor
143
144 def get_tensor(self, network) -> Tensor:
145 if self.is_managed(network):

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