Exemple #1
0
 def __init__(self, input_dataset, sleep_microseconds):
     self._input_dataset = input_dataset
     self._sleep_microseconds = sleep_microseconds
     variant_tensor = gen_experimental_dataset_ops.experimental_sleep_dataset(
         self._input_dataset._variant_tensor,  # pylint: disable=protected-access
         self._sleep_microseconds,
         **self._flat_structure)
     super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
Exemple #2
0
 def __init__(self, input_dataset, sleep_microseconds):
   self._input_dataset = input_dataset
   self._sleep_microseconds = sleep_microseconds
   variant_tensor = gen_experimental_dataset_ops.experimental_sleep_dataset(
       self._input_dataset._variant_tensor,  # pylint: disable=protected-access
       self._sleep_microseconds,
       **dataset_ops.flat_structure(self))
   super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
Exemple #3
0
 def _as_variant_tensor(self):
   return gen_experimental_dataset_ops.experimental_sleep_dataset(
       self._input_dataset._as_variant_tensor(),  # pylint: disable=protected-access
       self._sleep_microseconds,
       **dataset_ops.flat_structure(self))
Exemple #4
0
 def _as_variant_tensor(self):
     return gen_experimental_dataset_ops.experimental_sleep_dataset(
         self._input_dataset._as_variant_tensor(),  # pylint: disable=protected-access
         self._sleep_microseconds,
         **dataset_ops.flat_structure(self))