Home
last modified time | relevance | path

Searched refs:partition_graph (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_data.py776 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,
Dsession_debug_testlib.py247 for partition_graph in run_metadata.partition_graphs:
248 for node_def in partition_graph.node:
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc650 const GraphDef& partition_graph) { in AddPartitionGraph() argument
651 partition_graphs_.push_back(partition_graph); in AddPartitionGraph()
716 const GraphDef& partition_graph) { in AddPartitionGraph() argument
718 *graph_def = partition_graph; in AddPartitionGraph()
789 const GraphDef& partition_graph) { in AddPartitionGraph() argument
791 *graph_def = partition_graph; in AddPartitionGraph()
Dmessage_wrappers.h483 virtual void AddPartitionGraph(const GraphDef& partition_graph) = 0;
518 void AddPartitionGraph(const GraphDef& partition_graph) override;
551 void AddPartitionGraph(const GraphDef& partition_graph) override;
578 void AddPartitionGraph(const GraphDef& partition_graph) override;
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc227 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/
Ddirect_session.cc1246 std::unique_ptr<Graph>& partition_graph = iter->second; in CreateExecutors() local
1286 optimizer.Optimize(lib, options_.env, device, &partition_graph, in CreateExecutors()
1294 debug_options, partition_graph.get(), params.device)); in CreateExecutors()
1299 partition_graph.get())); in CreateExecutors()
1301 item->graph = partition_graph.get(); in CreateExecutors()
1306 executor_type, params, std::move(partition_graph), &item->executor)); in CreateExecutors()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto269 repeated GraphDef partition_graph = 4; field