Searched refs:get_next_ops (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | serialization_integration_test.py | 42 get_next_ops = []
47 get_next_ops.append(get_next_op)
49 return init_ops, get_next_ops, saver
60 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
65 output = self.evaluate(get_next_ops)
71 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
76 output = self.evaluate(get_next_ops)
|
| D | shuffle_dataset_serialization_test.py | 131 get_next_ops = [it.get_next() for it in iterators]
141 expected = [self.evaluate(get_next_ops) for _ in range(num_outputs)]
143 actual = [self.evaluate(get_next_ops) for _ in range(num_outputs)]
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 852 input_specs, get_next_ops): argument
857 return input_specs, get_next_ops
863 assert len(get_next_ops) == tpu_assignment.num_towers
869 get_next_ops[i] = [core_id_constant] + list(get_next_ops[i])
875 return input_specs, get_next_ops
|