parameters: (a: Argument, fn: Callable[[Node], Argument])
return_statement: return map_aggregate(a, lambda x: fn(x) if isinstance(x, Node) else x)
docstring: Apply fn to each Node appearing arg. arg may be a list, tuple, slice, or dict with string keys.
docstring_summary: Apply fn to each Node appearing arg. arg may be a list, tuple, slice, or dict with string keys.
docstring_tokens: Apply fn to each Node appearing arg . arg may be a list tuple slice or dict with string keys .
function: def map_arg(a: Argument, fn: Callable[[Node], Argument]) -> Argument:
      """
      Apply fn to each Node appearing arg. arg may be a list, tuple, slice, or dict with string keys.
      """
      assert callable(fn), "torch.fx.map_arg(a, fn): fn must be a callable"
      return map_aggregate(a, lambda x: fn(x) if isinstance(x, Node) else x)
function_tokens: def map_arg ( a : Argument , fn : Callable [ [ Node ] , Argument ] ) -> Argument : assert callable ( fn ) , "torch.fx.map_arg(a, fn): fn must be a callable" return map_aggregate ( a , lambda x : fn ( x ) if isinstance ( x , Node ) else x )