Home
last modified time | relevance | path

Searched refs:partition_graphs (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py169 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 …]
Dsession_debug_file_test.py93 len(run_metadata.partition_graphs))
96 run_dump_root, partition_graphs=run_metadata.partition_graphs)
Ddebug_graph_reconstruction_test.py94 non_debug_graph_defs = run_metadata.partition_graphs
105 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
119 run_metadata.partition_graphs[i]))
Dsession_debug_multi_gpu_test.py81 self._dump_root, partition_graphs=run_metadata.partition_graphs)
Ddebug_data.py469 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
Dsource_utils_test.py123 self.dump_root, partition_graphs=run_metadata.partition_graphs)
247 self.dump_root, partition_graphs=run_metadata.partition_graphs)
Ddebug_data_test.py222 partition_graphs=[graph_cpu_0, graph_gpu_0, graph_gpu_1])
266 partition_graphs=[graph_cpu_0, graph_gpu_0, graph_gpu_1])
Ddebug_gradients_test.py355 dump_dir, partition_graphs=run_metadata.partition_graphs)
/external/tensorflow/tensorflow/python/debug/wrappers/
Dlocal_cli_wrapper.py325 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/
Dtensorflow.-run-metadata.pbtxt20 name: "partition_graphs"
36 name: "partition_graphs"
Dtensorflow.-run-metadata.-function-graphs.pbtxt6 name: "partition_graphs"
/external/tensorflow/tensorflow/core/common_runtime/
Doptimization_registry.cc51 if (options.partition_graphs) { in RunGrouping()
52 for (auto& part : *options.partition_graphs) { in RunGrouping()
Dlower_if_while.cc45 if (options.partition_graphs != nullptr) { in Run()
Doptimization_registry.h59 std::unordered_map<string, std::unique_ptr<Graph>>* partition_graphs = member
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc178 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()
Dmessage_wrappers_test.cc162 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/
Dwhile_gradients_test.cc226 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/
Danalyzer_cli_test.py633 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/
Dmkl_tfconversion_pass.cc423 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/
Dcond_v2_test.py680 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)
Dcontrol_flow_ops_py_test.py677 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/
Dsummary_ops_v2.py1004 partition_graphs=data.partition_graphs)
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto522 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/
Dbackend_test.py1651 self.assertGreater(len(run_metadata.partition_graphs), 0)
1659 self.assertEqual(len(run_metadata.partition_graphs), 0)
/external/tensorflow/tensorflow/python/training/
Dmonitored_session_test.py1807 self.assertGreater(len(hook_b.run_metadata_list[0].partition_graphs), 0)
1855 self.assertGreater(len(hook.run_metadata_list[0].partition_graphs), 0)

12