Home
last modified time | relevance | path

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

1234

/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(
143 iterator = dataset_ops.make_one_shot_iterator(
172 iterator = dataset_ops.make_one_shot_iterator(dataset)
214 iterator = dataset_ops.make_one_shot_iterator(dataset)
407 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
408 iterator_4 = dataset_ops.make_one_shot_iterator(dataset_4)
466 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
[all …]
Diterator_checkpoint_test.py43 ) else dataset_ops.make_one_shot_iterator(dataset)
64 ) else dataset_ops.make_one_shot_iterator(dataset)
68 ) else dataset_ops.make_one_shot_iterator(dataset)
73 ) else dataset_ops.make_one_shot_iterator(dataset_2)
96 ) else dataset_ops.make_one_shot_iterator(dataset)
Dshuffle_test.py121 iterator = dataset_ops.make_one_shot_iterator(
198 iterator = dataset_ops.make_one_shot_iterator(dataset)
228 iterators = [dataset_ops.make_one_shot_iterator(dataset)
Diterator_cluster_test.py51 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
67 iterator_3 = dataset_ops.make_one_shot_iterator(dataset_3)
/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)
151 iterator = dataset_ops.make_one_shot_iterator(dataset)
255 iterator = dataset_ops.make_one_shot_iterator(dataset)
284 iterator = dataset_ops.make_one_shot_iterator(dataset)
321 iterator = dataset_ops.make_one_shot_iterator(dataset)
356 iterator = dataset_ops.make_one_shot_iterator(dataset)
385 iterator = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Doptimize_benchmark.py54 iterator = dataset_ops.make_one_shot_iterator(dataset)
95 iterator = dataset_ops.make_one_shot_iterator(dataset)
136 iterator = dataset_ops.make_one_shot_iterator(dataset)
Dautotune_benchmark.py51 iterator = dataset_ops.make_one_shot_iterator(dataset)
90 iterator = dataset_ops.make_one_shot_iterator(dataset)
128 iterator = dataset_ops.make_one_shot_iterator(dataset)
188 iterator = dataset_ops.make_one_shot_iterator(dataset)
Dchoose_fastest_benchmark.py67 iterator = dataset_ops.make_one_shot_iterator(dataset)
83 iterator = dataset_ops.make_one_shot_iterator(dataset)
Dmap_vectorization_benchmark.py124 unoptimized_next = dataset_ops.make_one_shot_iterator(
130 optimized_next = dataset_ops.make_one_shot_iterator(
Dmap_and_batch_benchmark.py146 chained_iterator = dataset_ops.make_one_shot_iterator(chained_dataset)
171 fused_iterator = dataset_ops.make_one_shot_iterator(fused_dataset)
Drejection_resample_benchmark.py45 get_next = dataset_ops.make_one_shot_iterator(dataset).get_next()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dprefetch_to_device_test.py42 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
68 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
92 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
121 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
Dcopy_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)
120 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
145 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
170 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
201 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
234 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dshuffle_dataset_serialization_test.py130 iterators = [ds.make_one_shot_iterator(), ds.make_one_shot_iterator()]
/external/tensorflow/tensorflow/python/distribute/
Dinput_ops_test.py93 iterator = dataset.make_one_shot_iterator()
142 iterator = dataset.make_one_shot_iterator()
175 iterator = dataset.make_one_shot_iterator()
213 iterator = dataset.make_one_shot_iterator()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_eager_test.py184 iterator = dataset_ops.make_one_shot_iterator(dataset)
187 validation_iterator = dataset_ops.make_one_shot_iterator(validation_dataset)
280 iterator = dataset_ops.make_one_shot_iterator(dataset)
Dtraining_dataset_test.py66 iterator = dataset_ops.make_one_shot_iterator(dataset)
127 iterator = dataset_ops.make_one_shot_iterator(dataset)
571 iterator = dataset_ops.make_one_shot_iterator(dataset)
580 iterator = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/
Drnn_ptb_graph_test.py85 inputs = tf.compat.v1.data.make_one_shot_iterator(dataset).get_next()
127 (inputs, labels) = tf.compat.v1.data.make_one_shot_iterator(
/external/tensorflow/tensorflow/python/data/
D__init__.py28 from tensorflow.python.data.ops.dataset_ops import make_one_shot_iterator
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.pbtxt36 name: "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/contrib/mixed_precision/python/
Dloss_scale_optimizer_test.py135 itr = dataset_ops.make_one_shot_iterator(dataset)
185 itr = dataset_ops.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/lite/tutorials/
Dmnist_tflite.py37 input_data = tf.compat.v1.data.make_one_shot_iterator(dataset.test(
/external/tensorflow/tensorflow/contrib/ignite/python/tests/
Dignite_dataset_test.py69 it = compat.v1.data.make_one_shot_iterator(dataset)
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py197 iterator = dataset.make_one_shot_iterator()
211 iterator = dataset.make_one_shot_iterator()

1234