message: Enable OneDeviceStrategy in the distributed embedding test.
  Only the GPU- or TPU-enabled strategies are covered by keras_embedding_model_correctness_test.py right now.  It is important to cover the OneDevice distribution strategy as well to make sure its API is tested.
diff: @@ -79,11 +79,14 @@ def all_strategies_excluding_tpu_and_input_config_combinations():
  def strategies_for_embedding_models():
  """Returns distribution strategies to test for embedding models.
  - Since embedding models take longer to train, we disregard OneDeviceStrategy
  - and DefaultStrategy in order to prevent testing timeouts.
  + Since embedding models take longer to train, we disregard DefaultStrategy
  + in order to prevent testing timeouts.
  """
  - return [s for s in all_strategies if s.required_tpu or s.required_gpus]
  + return [
  + s for s in all_strategies if s.required_tpu or s.required_gpus or
  + s is strategy_combinations.one_device_strategy
  + ]
  def test_combinations_for_embedding_model():
  