message: Register experimental sleep dataset op to GPU.
diff: @@ -135,6 +135,12 @@ class SleepDatasetOp : public UnaryDatasetOpKernel {
  REGISTER_KERNEL_BUILDER(Name("ExperimentalSleepDataset").Device(DEVICE_CPU),
  SleepDatasetOp);
  +REGISTER_KERNEL_BUILDER(Name("ExperimentalSleepDataset")
  + .Device(DEVICE_GPU)
  + .HostMemory("sleep_microseconds")
  + .HostMemory("input_dataset")
  + .HostMemory("handle"),
  + SleepDatasetOp);
  } // namespace
  } // namespace data
  