def __init__(self, input_dataset): self._input_dataset = input_dataset temp_variant_tensor = gen_dataset_ops.prefetch_dataset( input_dataset._variant_tensor, buffer_size=1, **dataset_ops.flat_structure(self)) variant_tensor = gen_dataset_ops.model_dataset( temp_variant_tensor, **dataset_ops.flat_structure(self)) super(_TestDataset, self).__init__(input_dataset, variant_tensor)
def __init__(self, input_dataset): self._input_dataset = input_dataset temp_variant_tensor = gen_dataset_ops.prefetch_dataset( input_dataset._variant_tensor, buffer_size=1, **self._flat_structure) variant_tensor = gen_dataset_ops.model_dataset(temp_variant_tensor, **self._flat_structure) super(_TestDataset, self).__init__(input_dataset, variant_tensor)
def _as_variant_tensor(self): return gen_dataset_ops.model_dataset( self._input_dataset._as_variant_tensor(), # pylint: disable=protected-access **dataset_ops.flat_structure(self))
def _as_variant_tensor(self): return gen_dataset_ops.model_dataset( self._input_dataset._as_variant_tensor(), # pylint: disable=protected-access **dataset_ops.flat_structure(self))