Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Ddataset_serialization_test_base.py41 def remove_variants(get_next_op): argument
52 return nest.map_structure(_remove_variant, get_next_op)
268 init_op, get_next_op = self._get_iterator_ops_from_collection(
270 get_next_op = remove_variants(get_next_op)
275 actual.append(sess.run(get_next_op))
278 sess.run(get_next_op)
331 _, get_next_op, saver = self._build_graph(
333 get_next_op = remove_variants(get_next_op)
337 actual.append(sess.run(get_next_op))
340 sess.run(get_next_op)
[all …]
Dserialization_integration_test.py45 init_op, get_next_op = self._build_input_pipeline(name, num_outputs)
47 get_next_ops.append(get_next_op)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_cluster_test.py58 get_next_op = remote_it.get_next()
62 sess.run(get_next_op)
Diterator_test.py801 get_next_op = iterator.get_next()
804 return init_op, get_next_op, save_op, restore_op
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_support.py752 get_next_op = self._iterator.get_next()
753 self._get_next_ops.append(get_next_op)
754 ctrl_deps.extend(get_next_op)