Fix imports for testing in OSS mode (pytorch#2684)

Summary:
- Fix imports for testing in OSS mode

Pull Request resolved: pytorch#2684

Reviewed By: sryap, spcyppt

Differential Revision: D58198181

Pulled By: q10

fbshipit-source-id: 84244dfb0b4e4c847ac9a2860660e00b9fcdd668

fbgemm_gpu/test/tbe/training/backward_dense_test.py

@@ -14,10 +14,6 @@
 import hypothesis.strategies as st
 import numpy as np
 import torch
-from deeplearning.fbgemm.fbgemm_gpu.test.tbe.common import (
-    format_ref_tensors_in_mixed_B_layout,
-    gen_mixed_B_batch_sizes,
-)
 from fbgemm_gpu.split_embedding_configs import SparseType
 from fbgemm_gpu.split_embedding_utils import (
     b_indices,
@@ -32,7 +28,11 @@
 from hypothesis import assume, given, HealthCheck, settings, Verbosity
 
 from .. import common  # noqa E402
-from ..common import open_source
+from ..common import (
+    format_ref_tensors_in_mixed_B_layout,
+    gen_mixed_B_batch_sizes,
+    open_source,
+)
 
 if open_source:
     # pyre-ignore[21]