| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | dataset_checkpoint_test.py | 66 def _build_graph(start, stop): function
80 init_op, get_next, save_op, _ = _build_graph(start, stop)
89 init_op, get_next, _, restore_op = _build_graph(start, stop)
100 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
115 def _build_graph(start, stop, num_epochs): function
131 init_op, get_next, save_op, _ = _build_graph(start, stop, num_epochs)
162 def _build_graph(start, stop): function
177 init_op, get_next, save_op, _ = _build_graph(start, stop)
188 init_op, get_next, _, restore_op = _build_graph(start, stop_1)
201 def _build_graph(start, stop): function
[all …]
|
| D | map_test.py | 315 def _build_graph(): function
324 captured_iter, ds = _build_graph()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | range_dataset_serialization_test.py | 58 def _build_graph(start, stop): function
72 init_op, get_next, save_op, _ = _build_graph(start, stop)
81 init_op, get_next, _, restore_op = _build_graph(start, stop)
92 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
|
| D | serialization_integration_test.py | 40 def _build_graph(self, num_pipelines, num_outputs): member in SerializationIntegrationTest
60 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
71 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
|
| D | dataset_serialization_test_base.py | 331 _, get_next_op, saver = self._build_graph(
426 init_op, get_next_op, saver = self._build_graph(
527 init_op, get_next_op, saver = self._build_graph(
596 def _build_graph(self, ds_fn, sparse_tensors=False): member in DatasetSerializationTestBase
|
| /external/tensorflow/tensorflow/contrib/mixed_precision/python/ |
| D | loss_scale_optimizer_test.py | 41 def _build_graph(self, lr, init_val, loss_scale_opt_fn=None): member in LossScaleOptimizerTest
61 x, loss, opt = self._build_graph(lr, init_val)
84 x, loss, opt = self._build_graph(lr, init_val, loss_scale_opt_fn)
106 x, loss, opt = self._build_graph(lr, init_val, loss_scale_opt_fn)
122 x, loss, opt = self._build_graph(lr, init_val)
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | shape_test.py | 66 def _build_graph(self, x, batch_ndims, event_ndims, expand_batch_dim): member in MakeBatchReadyTest
81 self._build_graph(
96 self._build_graph(x, batch_ndims, event_ndims, expand_batch_dim))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | fifo_queue_test.py | 1638 def _build_graph(self): member in FIFOQueueBenchmark
1665 init, output = self._build_graph()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 1142 def _build_graph(): member in SessionTest
1176 SessionTest._build_graph()
1197 build_threads = [self.checkedThread(target=SessionTest._build_graph)
|