Method _get_weights

tensorrt_llm/parameter.py:255–263

(self, network)

                self._get_weights(network), name)

    def _get_weights(self, network) -> trt.Weights | Tensor | None:
        tensor = network.get_parameter_tensor(self)
        if self.is_managed(network):
            return tensor
        elif tensor is not None:
            tensor.producer.__class__ = trt.IConstantLayer
            return tensor.producer.weights
        else:
            return None

    def _regularize_value(self, value):
        if isinstance(value, np.ndarray):