| /external/tensorflow/tensorflow/python/debug/lib/ |
| session_debug_file_test.py | 98 len(run_metadata.partition_graphs))
101 run_dump_root, partition_graphs=run_metadata.partition_graphs)
|
| session_debug_multi_gpu_test.py | 81 self._dump_root, partition_graphs=run_metadata.partition_graphs)
|
| debug_graph_reconstruction_test.py | 92 non_debug_graph_defs = run_metadata.partition_graphs
103 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
117 run_metadata.partition_graphs[i]))
|
| session_debug_testlib.py | 168 len(run_metadata.partition_graphs))
170 self._dump_root, partition_graphs=run_metadata.partition_graphs,
210 len(run_metadata.partition_graphs))
213 self._dump_root, partition_graphs=run_metadata.partition_graphs)
246 for partition_graph in run_metadata.partition_graphs:
349 self.assertEqual(1, len(run_metadata.partition_graphs))
352 self._dump_root, partition_graphs=run_metadata.partition_graphs)
[all...] |
| debug_data.py | 468 def __init__(self, dump_root, partition_graphs=None, validate=True):
473 partition_graphs: A repeated field of GraphDefs representing the
495 self._load_all_device_dumps(partition_graphs, validate)
499 def _load_all_device_dumps(self, partition_graphs, validate):
517 self._load_partition_graphs(partition_graphs, validate)
750 def _load_partition_graphs(self, partition_graphs, validate):
759 partition_graphs: A repeated field of GraphDefs representing the
771 if partition_graphs:
773 (partition_graph, None) for partition_graph in partition_graphs]
782 partition_graph = self._find_partition_graph(partition_graphs,
922 def partition_graphs(self): member in class:DebugDumpDir
[all...] |
| source_utils_test.py | 122 self.dump_root, partition_graphs=run_metadata.partition_graphs)
245 self.dump_root, partition_graphs=run_metadata.partition_graphs)
|
| 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])
|
| debug_gradients_test.py | 354 dump_dir, partition_graphs=run_metadata.partition_graphs)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| optimization_registry.h | 58 std::unordered_map<string, std::unique_ptr<Graph>>* partition_graphs = member in struct:tensorflow::GraphOptimizationPassOptions
|
| direct_session.cc | [all...] |
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| local_cli_wrapper.py | 309 partition_graphs = None
310 if request.run_metadata and request.run_metadata.partition_graphs:
311 partition_graphs = request.run_metadata.partition_graphs
313 partition_graphs = [request.client_graph_def]
323 self._dump_root, partition_graphs=partition_graphs)
|
| /external/tensorflow/tensorflow/cc/framework/ |
| while_gradients_test.cc | 226 ASSERT_EQ(run_metadata.partition_graphs().size(), 2);
227 for (const GraphDef& partition_graph : run_metadata.partition_graphs()) {
|
| /external/tensorflow/tensorflow/core/debug/ |
| debug_gateway_test.cc | 316 ASSERT_EQ(2, run_metadata.partition_graphs().size());
318 ASSERT_EQ(2, run_metadata.partition_graphs().size());
320 ASSERT_EQ(1, run_metadata.partition_graphs().size());
463 ASSERT_EQ(1, run_metadata.partition_graphs().size());
842 ASSERT_EQ(2, run_metadata.partition_graphs().size());
844 ASSERT_EQ(2, run_metadata.partition_graphs().size());
846 ASSERT_EQ(1, run_metadata.partition_graphs().size());
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.cc | 171 std::unordered_map<string, std::unique_ptr<Graph>> partition_graphs; local
180 partition_graphs.emplace(partition.first, std::move(device_graph));
185 optimization_options.partition_graphs = &partition_graphs;
195 for (auto& p : partition_graphs) {
|
| message_wrappers_test.cc | 162 response.metadata().partition_graphs(0).versions().producer());
164 response.metadata().partition_graphs(0).versions().min_consumer());
|
| /external/tensorflow/tensorflow/core/graph/ |
| mkl_tfconversion_pass.cc | 415 if (options.graph == nullptr && options.partition_graphs == nullptr) {
432 // options.partition_graphs.
433 for (auto& pg : *options.partition_graphs) {
|
| mkl_layout_pass.cc | [all...] |
| /external/tensorflow/tensorflow/python/debug/cli/ |
| analyzer_cli_test.py | 631 cls._dump_root, partition_graphs=run_metadata.partition_graphs)
[all...] |
| /external/tensorflow/tensorflow/python/client/ |
| session_test.py | [all...] |
| /external/tensorflow/tensorflow/python/training/ |
| monitored_session_test.py | [all...] |
