move Lp planner specific types into fb subfolder

Summary: Moving types into fb

Reviewed By: micrain

Differential Revision: D72881096

fbshipit-source-id: cc9035f8ca2c8c8b47f0dccb6e91cd116a713aff

Author: mserturk

torchrec/distributed/planner/types.py

@@ -342,39 +342,6 @@ def __str__(self) -> str:
         return f"Shard size: {tuple(self.size)}, offset: {tuple(self.offset)}, storage: {str(self.storage)}, perf: {str(self.perf)}, rank: {self.rank}"
 
 
-@dataclass
-class Rank:
-    key: str
-    hbm: float
-    ddr: float
-    node_key: str
-
-
-@dataclass
-class ShardInfo:
-    key: str
-    sharding_type: str
-    compute_kernel: str
-    table_name: str
-    rank_key: str  # either a number or DUMMY_RANK_KEY
-    shard: Shard
-    module_fqn: str
-    _index_in_shards: int
-    _index_in_search_space: int
-
-    def __hash__(self) -> int:
-        return hash(
-            (
-                self.key,
-                self.sharding_type,
-                self.compute_kernel,
-                self.table_name,
-                self.shard,
-                self.module_fqn,
-            )
-        )
-
-
 class ShardingOption:
     """
     One way of sharding an embedding table. In the enumerator, we generate