Searched refs:client_graph (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| D | graph_executor.cc | 423 const GraphExecutor::ClientGraph& client_graph) { in ImportAndCompileClientGraph() argument
425 loaded_client_graph->name = client_graph.name; in ImportAndCompileClientGraph()
433 auto module, ImportClientGraphToMlirModule(client_graph, &context)); in ImportAndCompileClientGraph()
435 LOG(INFO) << "TFRT finished importing client graph (" << &client_graph in ImportAndCompileClientGraph()
437 << " ms. Client graph name: " << client_graph.name; in ImportAndCompileClientGraph()
456 LOG(INFO) << "TFRT finished compiling client graph (" << &client_graph in ImportAndCompileClientGraph()
458 << " ms. Client graph name: " << client_graph.name; in ImportAndCompileClientGraph()
465 const GraphExecutor::ClientGraph& client_graph, in LoadClientGraph() argument
467 LOG(INFO) << "TFRT loading client graph (" << &client_graph << ") " in LoadClientGraph()
468 << client_graph.name; in LoadClientGraph()
[all …]
|
| D | graph_executor.h | 142 ImportAndCompileClientGraph(const GraphExecutor::ClientGraph& client_graph);
153 const GraphExecutor::ClientGraph& client_graph,
156 ImportClientGraphToMlirModule(const GraphExecutor::ClientGraph& client_graph,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 78 std::unique_ptr<ClientGraph> client_graph, in ReffedClientGraph() argument
85 client_graph_before_register_(std::move(client_graph)), in ReffedClientGraph()
317 PartitionOptions popts, ClientGraph* client_graph,
350 std::unique_ptr<ClientGraph> client_graph; in RegisterPartitions() local
351 std::swap(client_graph_before_register_, client_graph); in RegisterPartitions()
354 popts.flib_def = client_graph->flib_def.get(); in RegisterPartitions()
355 Status s = DoBuildPartitions(popts, client_graph.get(), &graph_defs); in RegisterPartitions()
431 PartitionOptions popts, ClientGraph* client_graph, in DoBuildPartitions() argument
435 cost_model.InitFromGraph(client_graph->graph); in DoBuildPartitions()
438 SlackAnalysis sa(&client_graph->graph, &cost_model); in DoBuildPartitions()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1606 std::unique_ptr<ClientGraph> client_graph; in CreateGraphs() local
1621 &temp_exec_state_holder, &client_graph)); in CreateGraphs()
1626 execution_state->BuildGraph(subgraph_options, &client_graph)); in CreateGraphs()
1628 *collective_graph_key = client_graph->collective_graph_key; in CreateGraphs()
1631 client_graph->feed_types.size()) { in CreateGraphs()
1636 client_graph->feed_types.size()); in CreateGraphs()
1639 client_graph->fetch_types.size()) { in CreateGraphs()
1644 client_graph->fetch_types.size()); in CreateGraphs()
1690 TF_RETURN_IF_ERROR(Partition(popts, &client_graph->graph, &partitions)); in CreateGraphs()
1714 new Graph(client_graph->flib_def.get())); in CreateGraphs()
[all …]
|