| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | session_debug_testlib.py | 164 len(run_metadata.partition_graphs))
166 self._dump_root, partition_graphs=run_metadata.partition_graphs,
206 len(run_metadata.partition_graphs))
209 self._dump_root, partition_graphs=run_metadata.partition_graphs)
242 for partition_graph in run_metadata.partition_graphs:
345 self.assertEqual(1, len(run_metadata.partition_graphs))
348 self._dump_root, partition_graphs=run_metadata.partition_graphs)
408 self._dump_root, partition_graphs=run_metadata.partition_graphs)
412 len(run_metadata.partition_graphs))
487 len(run_metadata.partition_graphs))
[all …]
|
| D | session_debug_file_test.py | 89 len(run_metadata.partition_graphs))
92 run_dump_root, partition_graphs=run_metadata.partition_graphs)
|
| D | debug_graph_reconstruction_test.py | 89 non_debug_graph_defs = run_metadata.partition_graphs
100 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
114 run_metadata.partition_graphs[i]))
|
| D | session_debug_multi_gpu_test.py | 77 self._dump_root, partition_graphs=run_metadata.partition_graphs)
|
| D | debug_grappler_test.py | 84 self._dump_root, partition_graphs=run_metadata.partition_graphs,
|
| D | debug_data.py | 464 def __init__(self, dump_root, partition_graphs=None, validate=True): argument
491 self._load_all_device_dumps(partition_graphs, validate)
495 def _load_all_device_dumps(self, partition_graphs, validate): argument
513 self._load_partition_graphs(partition_graphs, validate)
796 def _find_partition_graph(self, partition_graphs, device_name): argument
797 if partition_graphs is None:
800 for graph_def in partition_graphs:
919 def partition_graphs(self): member in DebugDumpDir
|
| D | source_utils_test.py | 180 self.dump_root, partition_graphs=run_metadata.partition_graphs)
347 self.dump_root, partition_graphs=run_metadata.partition_graphs)
|
| D | debug_data_test.py | 217 partition_graphs=[graph_cpu_0, graph_gpu_0, graph_gpu_1])
259 partition_graphs=[graph_cpu_0, graph_gpu_0, graph_gpu_1])
|
| D | debug_gradients_test.py | 351 dump_dir, partition_graphs=run_metadata.partition_graphs)
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | local_cli_wrapper.py | 318 partition_graphs = None
319 if request.run_metadata and request.run_metadata.partition_graphs:
320 partition_graphs = request.run_metadata.partition_graphs
322 partition_graphs = [request.client_graph_def]
332 self._dump_root, partition_graphs=partition_graphs)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-metadata.pbtxt | 20 name: "partition_graphs"
43 name: "partition_graphs"
|
| D | tensorflow.-run-metadata.-function-graphs.pbtxt | 6 name: "partition_graphs"
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | optimization_registry.cc | 44 if (options.partition_graphs) { in RunGrouping()
45 for (auto& part : *options.partition_graphs) { in RunGrouping()
|
| D | optimization_registry.h | 65 std::unordered_map<string, std::unique_ptr<Graph>>* partition_graphs = member
|
| D | mkl_tfconversion_pass.cc | 425 if (options.graph == nullptr && options.partition_graphs == nullptr) { in Run()
452 for (auto& pg : *options.partition_graphs) { in Run()
|
| D | lower_functional_ops.cc | 101 if (options.partition_graphs != nullptr) { in Run()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 192 std::unordered_map<string, std::unique_ptr<Graph>> partition_graphs; in InitItem() local
201 partition_graphs.emplace(partition.first, std::move(device_graph)); in InitItem()
206 optimization_options.partition_graphs = &partition_graphs; in InitItem()
216 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 | 72 for (auto& it : *options.partition_graphs) { in Run()
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | analyzer_cli_test.py | 644 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
1646 partition_graphs=run_metadata.partition_graphs)
1678 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
1758 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
2075 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | remapper_test.py | 116 graph = metadata.partition_graphs[0]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | summary_ops_v2.py | 1272 partition_graphs=data.partition_graphs)
|
| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | cond_v2_test.py | 1555 self.assertTrue(len(run_metadata.partition_graphs) >= 2)
1704 self.assertGreaterEqual(len(run_metadata.partition_graphs), 2)
1757 for graph in run_metadata.partition_graphs:
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 802 repeated GraphDef partition_graphs = 3; field
806 repeated GraphDef partition_graphs = 1; field
813 // The main use cases of the post_optimization_graph and the partition_graphs
|