Searched refs:partition_graph (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_data.py | 771 partition_graph = None
773 partition_graph = _load_graph_def_from_event_file(
780 partition_graph = self._find_partition_graph(client_partition_graphs,
783 if partition_graph:
784 partition_graphs_and_device_names.append((partition_graph,
787 for partition_graph, maybe_device_name in partition_graphs_and_device_names:
788 debug_graph = debug_graphs.DebugGraph(partition_graph,
|
| D | session_debug_testlib.py | 242 for partition_graph in run_metadata.partition_graphs:
243 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()
739 const GraphDef& partition_graph) { in AddPartitionGraph() argument
741 *graph_def = partition_graph; in AddPartitionGraph()
816 const GraphDef& partition_graph) { in AddPartitionGraph() argument
818 *graph_def = partition_graph; in AddPartitionGraph()
|
| D | message_wrappers.h | 495 virtual void AddPartitionGraph(const GraphDef& partition_graph) = 0;
531 void AddPartitionGraph(const GraphDef& partition_graph) override;
565 void AddPartitionGraph(const GraphDef& partition_graph) override;
593 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 | 1375 std::unique_ptr<Graph>& partition_graph = iter->second; in CreateExecutors() local
1417 optimizer.Optimize(lib, options_.env, device, &partition_graph, in CreateExecutors()
1425 debug_options, partition_graph.get(), params.device)); in CreateExecutors()
1430 partition_graph.get())); in CreateExecutors()
1436 NewExecutor(executor_type, params, *partition_graph, &item->executor)); in CreateExecutors()
1439 item->graph = std::move(partition_graph); in CreateExecutors()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 1431 for partition_graph in run_metadata.partition_graphs:
1432 for node in partition_graph.node:
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 298 repeated GraphDef partition_graph = 4; field
|