Method set_parameter_tensor

tensorrt_llm/network.py:179–181

(self, param, tensor)

        return self.parameter_tensors.get(param, None)

    def set_parameter_tensor(self, param, tensor):
        assert param not in self.parameter_tensors
        self.parameter_tensors[param] = tensor

    @property
    def dtype(self) -> trt.DataType: