| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | session_debug_testlib.py | 169 len(run_metadata.partition_graphs))
171 self._dump_root, partition_graphs=run_metadata.partition_graphs,
211 len(run_metadata.partition_graphs))
214 self._dump_root, partition_graphs=run_metadata.partition_graphs)
247 for partition_graph in run_metadata.partition_graphs:
350 self.assertEqual(1, len(run_metadata.partition_graphs))
353 self._dump_root, partition_graphs=run_metadata.partition_graphs)
413 self._dump_root, partition_graphs=run_metadata.partition_graphs)
417 len(run_metadata.partition_graphs))
492 len(run_metadata.partition_graphs))
[all …]
|
| D | session_debug_file_test.py | 93 len(run_metadata.partition_graphs))
96 run_dump_root, partition_graphs=run_metadata.partition_graphs)
|
| D | debug_graph_reconstruction_test.py | 94 non_debug_graph_defs = run_metadata.partition_graphs
105 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
119 run_metadata.partition_graphs[i]))
|
| D | session_debug_multi_gpu_test.py | 81 self._dump_root, partition_graphs=run_metadata.partition_graphs)
|
| D | debug_data.py | 469 def __init__(self, dump_root, partition_graphs=None, validate=True): argument
496 self._load_all_device_dumps(partition_graphs, validate)
500 def _load_all_device_dumps(self, partition_graphs, validate): argument
518 self._load_partition_graphs(partition_graphs, validate)
801 def _find_partition_graph(self, partition_graphs, device_name): argument
802 if partition_graphs is None:
805 for graph_def in partition_graphs:
924 def partition_graphs(self): member in DebugDumpDir
|
| D | source_utils_test.py | 123 self.dump_root, partition_graphs=run_metadata.partition_graphs)
247 self.dump_root, partition_graphs=run_metadata.partition_graphs)
|
| D | debug_data_test.py | 222 partition_graphs=[graph_cpu_0, graph_gpu_0, graph_gpu_1])
266 partition_graphs=[graph_cpu_0, graph_gpu_0, graph_gpu_1])
|
| D | debug_gradients_test.py | 355 dump_dir, partition_graphs=run_metadata.partition_graphs)
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | local_cli_wrapper.py | 325 partition_graphs = None
326 if request.run_metadata and request.run_metadata.partition_graphs:
327 partition_graphs = request.run_metadata.partition_graphs
329 partition_graphs = [request.client_graph_def]
339 self._dump_root, partition_graphs=partition_graphs)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-metadata.pbtxt | 20 name: "partition_graphs"
36 name: "partition_graphs"
|
| D | tensorflow.-run-metadata.-function-graphs.pbtxt | 6 name: "partition_graphs"
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | optimization_registry.cc | 51 if (options.partition_graphs) { in RunGrouping()
52 for (auto& part : *options.partition_graphs) { in RunGrouping()
|
| D | lower_if_while.cc | 45 if (options.partition_graphs != nullptr) { in Run()
|
| D | optimization_registry.h | 59 std::unordered_map<string, std::unique_ptr<Graph>>* partition_graphs = member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 178 std::unordered_map<string, std::unique_ptr<Graph>> partition_graphs; in InitItem() local
187 partition_graphs.emplace(partition.first, std::move(device_graph)); in InitItem()
192 optimization_options.partition_graphs = &partition_graphs; in InitItem()
202 for (auto& p : partition_graphs) { in InitItem()
|
| D | message_wrappers_test.cc | 162 response.metadata().partition_graphs(0).versions().producer()); in CheckRunStepResponse()
164 response.metadata().partition_graphs(0).versions().min_consumer()); in CheckRunStepResponse()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients_test.cc | 226 ASSERT_EQ(run_metadata.partition_graphs().size(), 2); in TEST_F()
227 for (const GraphDef& partition_graph : run_metadata.partition_graphs()) { in TEST_F()
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | analyzer_cli_test.py | 633 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
1603 partition_graphs=run_metadata.partition_graphs)
1634 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
1714 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
2031 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_tfconversion_pass.cc | 423 if (options.graph == nullptr && options.partition_graphs == nullptr) { in Run()
445 for (auto& pg : *options.partition_graphs) { in Run()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 680 for graph in run_metadata.partition_graphs
689 for graph in run_metadata.partition_graphs
716 for graph in run_metadata.partition_graphs
726 for graph in run_metadata.partition_graphs
1099 self.assertTrue(len(run_metadata.partition_graphs) >= 2)
1166 self.assertTrue(len(run_metadata.partition_graphs) >= 2)
|
| D | control_flow_ops_py_test.py | 677 self.assertEqual(len(run_metadata.partition_graphs), 1)
683 g for g in run_metadata.partition_graphs
715 self.assertEqual(len(run_metadata.partition_graphs), 2)
743 self.assertEqual(len(run_metadata.partition_graphs), 2)
1656 self.assertEqual(len(run_metadata.partition_graphs), 1)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | summary_ops_v2.py | 1004 partition_graphs=data.partition_graphs)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 522 repeated GraphDef partition_graphs = 3; field
526 repeated GraphDef partition_graphs = 1; field
533 // The main use cases of the post_optimization_graph and the partition_graphs
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 1651 self.assertGreater(len(run_metadata.partition_graphs), 0)
1659 self.assertEqual(len(run_metadata.partition_graphs), 0)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1807 self.assertGreater(len(hook_b.run_metadata_list[0].partition_graphs), 0)
1855 self.assertGreater(len(hook.run_metadata_list[0].partition_graphs), 0)
|