def get_source_nodes_for_output_name(
        self,
        output_name: str,
    ) -> Optional[TorchNode]:
        """
        Retrieves the source node for a given MIL operation's output name.

        Parameters
        ----------
        output_name : str
            The name of the output.

        Returns
        -------
        Optional[TorchNode]
            The corresponding TorchNode if found, None otherwise.
        """
        return self.operation_output_name_to_node_map.get(output_name, None)

        Retrieves the source node for a given MIL operation's output name.

        Parameters
        ----------
        output_name : str
            The name of the output.

        Returns
        -------
        Optional[TorchNode]
            The corresponding TorchNode if found, None otherwise.
        
coremltools/models/ml_program/experimental/torch/debugging_utils.py