Home
last modified time | relevance | path

Searched refs:get_next_op (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Ddataset_serialization_test_base.py41 def remove_variants(get_next_op): argument
51 return nest.map_structure(_remove_variant, get_next_op)
228 init_op, get_next_op = self._get_iterator_ops_from_collection(
230 get_next_op = remove_variants(get_next_op)
236 actual.append(sess.run(get_next_op))
239 sess.run(get_next_op)
263 init_op, get_next_op, saver = self._build_graph(
265 get_next_op = remove_variants(get_next_op)
269 sess.run(get_next_op)
352 init_op, get_next_op = self._get_iterator_ops_from_collection(
[all …]
Dserialization_integration_test.py49 init_op, get_next_op = self._build_input_pipeline(name, num_outputs)
51 get_next_ops.append(get_next_op)
Dsnapshot_dataset_serialization_test.py180 init_op, get_next_op, saver = self._build_graph(ds_fn)
187 outputs.append(sess.run(get_next_op))
193 outputs.append(sess.run(get_next_op))
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_cluster_test.py61 get_next_op = remote_it.get_next()
65 sess.run(get_next_op)