| /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 | 93 non_debug_graph_defs = run_metadata.partition_graphs
104 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
118 run_metadata.partition_graphs[i]))
|
| D | session_debug_multi_gpu_test.py | 81 self._dump_root, partition_graphs=run_metadata.partition_graphs)
|
| D | debug_grappler_test.py | 88 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 | 184 self.dump_root, partition_graphs=run_metadata.partition_graphs)
351 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])
264 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 | 322 partition_graphs = None
323 if request.run_metadata and request.run_metadata.partition_graphs:
324 partition_graphs = request.run_metadata.partition_graphs
326 partition_graphs = [request.client_graph_def]
336 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 | 58 if (options.partition_graphs) { in RunGrouping()
59 for (auto& part : *options.partition_graphs) { in RunGrouping()
|
| D | optimization_registry.h | 59 std::unordered_map<string, std::unique_ptr<Graph>>* partition_graphs = member
|
| D | mkl_tfconversion_pass.cc | 426 if (options.graph == nullptr && options.partition_graphs == nullptr) { in Run()
453 for (auto& pg : *options.partition_graphs) { in Run()
|
| D | lower_functional_ops.cc | 82 if (options.partition_graphs != nullptr) { in Run()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 190 std::unordered_map<string, std::unique_ptr<Graph>> partition_graphs; in InitItem() local
199 partition_graphs.emplace(partition.first, std::move(device_graph)); in InitItem()
204 optimization_options.partition_graphs = &partition_graphs; in InitItem()
214 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/compiler/mlir/tensorflow/translate/ |
| D | mlir_roundtrip_pass.cc | 70 for (auto& it : *options.partition_graphs) { in Run()
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | analyzer_cli_test.py | 633 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
1629 partition_graphs=run_metadata.partition_graphs)
1660 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
1740 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
2057 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | summary_ops_v2.py | 1205 partition_graphs=data.partition_graphs)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 1554 self.assertTrue(len(run_metadata.partition_graphs) >= 2)
1703 self.assertGreaterEqual(len(run_metadata.partition_graphs), 2)
1756 for graph in run_metadata.partition_graphs:
|
| D | control_flow_ops_py_test.py | 723 self.assertEqual(len(run_metadata.partition_graphs), 1)
730 g for g in run_metadata.partition_graphs
762 self.assertLen(run_metadata.partition_graphs, 2)
794 self.assertLen(run_metadata.partition_graphs, 2)
824 self.assertEqual(len(run_metadata.partition_graphs), 2)
1940 self.assertEqual(len(run_metadata.partition_graphs), 1)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 716 repeated GraphDef partition_graphs = 3; field
720 repeated GraphDef partition_graphs = 1; field
727 // The main use cases of the post_optimization_graph and the partition_graphs
|