    def replace_input_tensor(self, old_tensor, new_tensor) -> None:
        """Replaces old_tensors in self._attrs["inputs"] with new_tensor.

        Parameters
        ----------
        old_tensor: Tensor
            The old tensor to be replaced.
        new_tensor: Tensor
            The new tensor.

        Returns
        -------
        None.
        """

        self._attrs["inputs"] = [
            new_tensor if tensor is old_tensor else tensor
            for tensor in self._attrs["inputs"]
        ]
Replaces old_tensors in self._attrs["inputs"] with new_tensor.

Parameters
----------
old_tensor: Tensor
The old tensor to be replaced.
new_tensor: Tensor
The new tensor.

Returns
-------
None.
python/aitemplate/compiler/base.py