Adding "enable_optimizer_offloading" and "optimizer_D" to the required function signatures through the flow (#5056)

Summary:

X-link: meta-pytorch/torchrec#3484

X-link: facebookresearch/FBGEMM#2066

**Context:**
Currently, we are enabling SSD optimizer offloading for the ssd tbe kernel

**In this diff:**
We retrieve the newly added parameters from the tbe config and pass it down to the tbe

Differential Revision: D85353134

Author: Raahul Kalyaan Jakka

fbgemm_gpu/fbgemm_gpu/tbe/ssd/training.py

@@ -186,6 +186,7 @@ def __init__(
         use_rowwise_bias_correction: bool = False,  # For Adam use
         optimizer_state_dtypes: dict[str, SparseType] = {},  # noqa: B006
         pg: Optional[dist.ProcessGroup] = None,
+        enable_optimizer_offloading: bool = False,
     ) -> None:
         super(SSDTableBatchedEmbeddingBags, self).__init__()