Home
last modified time | relevance | path

Searched refs:graph_out (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc169 Status BuildOutputGraph(Graph* graph_out, FunctionLibraryDefinition* library);
211 Graph* graph_out);
246 Status MakeSequencingNode(const string& subgraph_name, Graph* graph_out);
250 void ConnectSequencerToCallNode(Graph* graph_out);
315 Graph* graph_out, std::unordered_map<const Node*, Node*>* node_images);
320 Graph* graph_out);
357 Graph* graph_out,
364 Graph* graph_out);
543 Graph* graph_out) { in MakeSequencingNode() argument
553 sequencer_ = graph_out->AddNode(seq_def, &s); in MakeSequencingNode()
[all …]
Dencapsulate_subgraphs_pass.h78 std::unique_ptr<Graph>* graph_out, FunctionLibraryDefinition* library);
Dextract_outside_compilation_pass.cc2345 std::unique_ptr<Graph> graph_out; in ExtractOutsideCompilationForFunction() local
2369 /*reuse_existing_functions=*/true, &graph_out, fld)); in ExtractOutsideCompilationForFunction()
2373 for (Node* n : graph_out->nodes()) { in ExtractOutsideCompilationForFunction()
2408 graph_out.get(), n, host_compute_core, *cluster_deps); in ExtractOutsideCompilationForFunction()
2428 graph_out->AddControlEdge(iter->second, host_compute_node); in ExtractOutsideCompilationForFunction()
2434 Graph* g = (*has_outside_compilation) ? graph_out.get() : fbody->graph; in ExtractOutsideCompilationForFunction()
Dencapsulate_subgraphs_pass_test.cc508 std::unique_ptr<Graph> graph_out; in Encapsulate() local
512 &graph_out, lib_def.get()); in Encapsulate()
518 for (Node* n : graph_out->nodes()) { in Encapsulate()
534 graph_out.get(), flr, lib_def.get(), &modified); in Encapsulate()
538 graph_out->ToGraphDef(&graphdef_out); in Encapsulate()