def METHOD_NAME(self, mod: torch.nn.Module) -> str:
    """
    Allows trace-ability of non registered modules. This is typically used for Table Batched Embeddings
    made to look like nn.EmbeddingBags
    """
    if hasattr(mod, "_fx_path"):
        return mod._fx_path
    else:
        return super().METHOD_NAME(mod)