Method set_name

tensorrt_llm/parameter.py:246–253

(self, name: str, network)

        self.value = v

    def set_name(self, name: str, network):
        self._name = name
        if self.is_managed(network):
            self._get_weights(network).name = name
            return True
        else:
            return network.trt_network.set_weights_name(
                self._get_weights(network), name)

    def _get_weights(self, network) -> trt.Weights | Tensor | None:
        tensor = network.get_parameter_tensor(self)