Python prefetch_dataset示例

示例#1

文件： input_ops_test.py 项目： 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)

示例#2

文件： traverse_test.py 项目： 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)