Searched refs:partition_graph (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_data.py | 776 partition_graph = None
778 partition_graph = _load_graph_def_from_event_file(
785 partition_graph = self._find_partition_graph(client_partition_graphs,
788 if partition_graph:
789 partition_graphs_and_device_names.append((partition_graph,
792 for partition_graph, maybe_device_name in partition_graphs_and_device_names:
793 debug_graph = debug_graphs.DebugGraph(partition_graph,
|
| D | session_debug_testlib.py | 247 for partition_graph in run_metadata.partition_graphs:
248 for node_def in partition_graph.node:
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers.cc | 669 const GraphDef& partition_graph) { in AddPartitionGraph() argument
670 partition_graphs_.push_back(partition_graph); in AddPartitionGraph()
735 const GraphDef& partition_graph) { in AddPartitionGraph() argument
737 *graph_def = partition_graph; in AddPartitionGraph()
808 const GraphDef& partition_graph) { in AddPartitionGraph() argument
810 *graph_def = partition_graph; in AddPartitionGraph()
|
| D | message_wrappers.h | 495 virtual void AddPartitionGraph(const GraphDef& partition_graph) = 0;
530 void AddPartitionGraph(const GraphDef& partition_graph) override;
563 void AddPartitionGraph(const GraphDef& partition_graph) override;
590 void AddPartitionGraph(const GraphDef& partition_graph) override;
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients_test.cc | 227 for (const GraphDef& partition_graph : run_metadata.partition_graphs()) { in TEST_F() local
228 EXPECT_GE(partition_graph.node().size(), 1); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1358 std::unique_ptr<Graph>& partition_graph = iter->second; in CreateExecutors() local
1400 optimizer.Optimize(lib, options_.env, device, &partition_graph, in CreateExecutors()
1408 debug_options, partition_graph.get(), params.device)); in CreateExecutors()
1413 partition_graph.get())); in CreateExecutors()
1419 NewExecutor(executor_type, params, *partition_graph, &item->executor)); in CreateExecutors()
1422 item->graph = std::move(partition_graph); in CreateExecutors()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 1401 for partition_graph in run_metadata.partition_graphs:
1402 for node in partition_graph.node:
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 296 repeated GraphDef partition_graph = 4; field
|