class_docstr: A torch.fx graph transformation that is reversible.
  
  It must implement the [`~optimum.fx.optimization.ReversibleTransformation.transform`] and
  [`~optimum.fx.optimization.ReversibleTransformation.reverse`] methods, and be used as a callable.
class_docstr_tok: ["A","torch.fx","graph","transformation","that","is","reversible",".","It","must","implement","the","[","`","~optimum.fx.optimization.ReversibleTransformation.transform","`","]","and","[","`","~optimum.fx.optimization.ReversibleTransformation.reverse","`","]","methods",",","and","be","used","as","a","callable","."]
docstr: Args:
      graph_module (`torch.fx.GraphModule`):
          The module to transform.
  
  Returns:
      `torch.fx.GraphModule`:
          The reverse transformed module.
docstr_tok: ["Args",":","graph_module","(","`","torch.fx.GraphModule","`",")",":","The","module","to","transform",".","Returns",":","`","torch.fx.GraphModule","`",":","The","reverse","transformed","module","."]
code: def reverse(self, graph_module: "GraphModule") -> "GraphModule":
      """
      Args:
          graph_module (`torch.fx.GraphModule`):
              The module to transform.
  
      Returns:
          `torch.fx.GraphModule`:
              The reverse transformed module.
      """
      raise NotImplementedError(
          "The reverse transform method needs to be implemented."
      )
  
imports: import collections
  import copy
  import functools
  import itertools
  import operator
  import warnings
  from abc import ABC, abstractmethod
  from typing import TYPE_CHECKING, List
  import torch
  from transformers.file_utils import add_end_docstrings
  from transformers.utils.fx import _gen_constructor_wrapper
label_desc: Use image node_id 1 for calling the ReversibleTransformation obj's underlying member method code with example usage: obj.reverse(graph_module) without return types