class_docstr: Tracer with an extended set of leaf nn.Modules.
code: def is_leaf_module(
      self, m: torch.nn.Module, model_qualified_name: str
  ) -> bool:
      return (
          super().is_leaf_module(m, model_qualified_name)
          or m in self.leaf_modules
      )
  
file_path: fairscale/fairscale/experimental/nn/auto_shard.py
imports: import logging
  from typing import Dict, List, Set
  import torch
  import torch.fx
  from torch.fx.node import Node
label_desc: Use image node_id 2 for calling the _ExtendedLeafTracer obj's underlying member method code with example usage: obj.is_leaf_module(m, model_qualified_name) and returns: unknown