Home
last modified time | relevance | path

Searched refs:make_one_shot_iterator (Results 1 – 25 of 59) sorted by relevance

123

/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_test.py64 value = dataset_ops.make_one_shot_iterator(dataset).get_next()
78 dataset_ops.make_one_shot_iterator(dataset)
89 iterator = dataset_ops.make_one_shot_iterator(
116 iterator = dataset_ops.make_one_shot_iterator(
144 iterator = dataset_ops.make_one_shot_iterator(
173 iterator = dataset_ops.make_one_shot_iterator(dataset)
212 iterator = dataset_ops.make_one_shot_iterator(dataset)
411 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
412 iterator_4 = dataset_ops.make_one_shot_iterator(dataset_4)
469 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
[all …]
Diterator_cluster_test.py54 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
70 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
Dshuffle_test.py126 iterator = dataset_ops.make_one_shot_iterator(
177 iterator = dataset_ops.make_one_shot_iterator(dataset)
205 iterators = [dataset_ops.make_one_shot_iterator(dataset)
Dtest_base.py131 iterator = dataset_ops.make_one_shot_iterator(dataset)
Dplacement_test.py181 iterator = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py51 iterator = dataset_ops.make_one_shot_iterator(dataset)
76 iterator = dataset_ops.make_one_shot_iterator(dataset)
112 iterator = dataset_ops.make_one_shot_iterator(dataset)
125 iterator = dataset_ops.make_one_shot_iterator(dataset)
152 iterator = dataset_ops.make_one_shot_iterator(dataset)
256 iterator = dataset_ops.make_one_shot_iterator(dataset)
285 iterator = dataset_ops.make_one_shot_iterator(dataset)
322 iterator = dataset_ops.make_one_shot_iterator(dataset)
357 iterator = dataset_ops.make_one_shot_iterator(dataset)
386 iterator = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dprefetch_to_device_test.py46 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
71 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
95 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
123 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
Dcopy_to_device_test.py49 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
74 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
98 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
123 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
148 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
173 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
204 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
237 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
Dmap_and_batch_test.py162 iterator = dataset_ops.make_one_shot_iterator(dataset)
189 iterator = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dshuffle_dataset_serialization_test.py92 iterators = [ds.make_one_shot_iterator(), ds.make_one_shot_iterator()]
Dcheckpoint_input_pipeline_hook_test.py96 iterator = ds.make_one_shot_iterator()
/external/tensorflow/tensorflow/python/autograph/operators/
Dpy_builtins_test.py61 iterator = dataset_ops.make_one_shot_iterator(dataset)
72 iterator = dataset_ops.make_one_shot_iterator(dataset)
85 iterator = dataset_ops.make_one_shot_iterator(dataset)
237 iterator = dataset_ops.make_one_shot_iterator(dataset)
252 iterator = dataset_ops.make_one_shot_iterator(ds3)
276 iterator = dataset_ops.make_one_shot_iterator(ds2)
287 iterator = dataset_ops.make_one_shot_iterator(ds3)
583 iterator = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/data/
D__init__.py30 from tensorflow.python.data.ops.dataset_ops import make_one_shot_iterator
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dprefetching_ops.py219 def make_one_shot_iterator(self): member in _CopyToDeviceDataset
225 return super(_CopyToDeviceDataset, self).make_one_shot_iterator()
/external/tensorflow/tensorflow/lite/tutorials/
Dmnist_tflite.py38 input_data = tf.compat.v1.data.make_one_shot_iterator(dataset.test(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.pbtxt64 name: "make_one_shot_iterator"
Dtensorflow.data.-dataset.pbtxt95 name: "make_one_shot_iterator"
Dtensorflow.data.-t-f-record-dataset.pbtxt97 name: "make_one_shot_iterator"
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py195 iterator = dataset_ops.make_one_shot_iterator(dataset)
206 iterator = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/tpu/
Ddatasets.py161 source_iterator = dataset_ops.make_one_shot_iterator(source_dataset)
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50_graph_test.py107 images, labels = tf.data.make_one_shot_iterator(
/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy_test.py82 it = dataset_ops.make_one_shot_iterator(dataset)
Dcollective_all_reduce_strategy_test.py354 it = dataset_ops.make_one_shot_iterator(dataset)
453 it = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/keras/feature_column/
Dsequence_feature_column_integration_test.py96 features = dataset_ops.make_one_shot_iterator(ds).get_next()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_generator_v1.py326 generator = dataset_ops.make_one_shot_iterator(original_dataset)
464 return dataset_ops.make_one_shot_iterator(data), steps_per_epoch

123