Esempi in Python per prefetch_dataset

Esempio n. 1

File: input_ops_test.py Progetto: adit-chandra/tensorflow

 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)

Esempio n. 2

File: traverse_test.py Progetto: xutianming/tensorflow

 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)