class_docstr: A torch.fx graph transformation.
  
  It  must implement the [`~optimum.fx.optimization.ReversibleTransformation.transform`] method, and be used as a
  callable.
class_docstr_tok: ["A","torch.fx","graph","transformation",".","It","must","implement","the","[","`","~optimum.fx.optimization.ReversibleTransformation.transform","`","]","method",",","and","be","used","as","a","callable","."]
docstr: Args:
      graph_module (`torch.fx.GraphModule`):
          The module to transform.
  
  Returns:
      `torch.fx.GraphModule`:
          The transformed module.
docstr_tok: ["Args",":","graph_module","(","`","torch.fx.GraphModule","`",")",":","The","module","to","transform",".","Returns",":","`","torch.fx.GraphModule","`",":","The","transformed","module","."]
code: def transform(self, graph_module: "GraphModule") -> "GraphModule":
      """
      Args:
          graph_module (`torch.fx.GraphModule`):
              The module to transform.
  
      Returns:
          `torch.fx.GraphModule`:
              The transformed module.
      """
      raise NotImplementedError(
          "The 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 Transformation obj's underlying member method code with example usage: obj.transform(graph_module) without return types