Searched refs:client_graph (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1464 std::unique_ptr<ClientGraph> client_graph; in CreateGraphs() local
1480 &temp_exec_state_holder, &client_graph)); in CreateGraphs()
1485 execution_state->BuildGraph(subgraph_options, &client_graph)); in CreateGraphs()
1487 *collective_graph_key = client_graph->collective_graph_key; in CreateGraphs()
1490 client_graph->feed_types.size()) { in CreateGraphs()
1495 client_graph->feed_types.size()); in CreateGraphs()
1498 client_graph->fetch_types.size()) { in CreateGraphs()
1503 client_graph->fetch_types.size()); in CreateGraphs()
1545 popts.flib_def = &client_graph->graph.flib_def(); in CreateGraphs()
1549 TF_RETURN_IF_ERROR(Partition(popts, &client_graph->graph, &partitions)); in CreateGraphs()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 68 std::unique_ptr<ClientGraph> client_graph, in ReffedClientGraph() argument
75 client_graph_before_register_(std::move(client_graph)), in ReffedClientGraph()
307 PartitionOptions popts, ClientGraph* client_graph,
340 std::unique_ptr<ClientGraph> client_graph; in RegisterPartitions() local
341 std::swap(client_graph_before_register_, client_graph); in RegisterPartitions()
344 popts.flib_def = client_graph->flib_def.get(); in RegisterPartitions()
345 Status s = DoBuildPartitions(popts, client_graph.get(), &graph_defs); in RegisterPartitions()
421 PartitionOptions popts, ClientGraph* client_graph, in DoBuildPartitions() argument
425 cost_model.InitFromGraph(client_graph->graph); in DoBuildPartitions()
428 SlackAnalysis sa(&client_graph->graph, &cost_model); in DoBuildPartitions()
[all …]
|