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)
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)
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))