Home
last modified time | relevance | path

Searched refs:host_fbody (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc342 std::unique_ptr<FunctionBody> host_fbody; in TEST_F() local
348 *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, &host_fbody)); in TEST_F()
349 Graph *host_graph = host_fbody->graph; in TEST_F()
498 std::unique_ptr<FunctionBody> host_fbody; in TEST_F() local
505 &host_fbody)); in TEST_F()
506 Graph *host_graph = host_fbody->graph; in TEST_F()
648 std::unique_ptr<FunctionBody> host_fbody; in TEST_F() local
655 &host_fbody)); in TEST_F()
656 Graph *host_graph = host_fbody->graph; in TEST_F()
784 std::unique_ptr<FunctionBody> host_fbody; in TEST_F() local
[all …]
Dextract_outside_compilation_pass.cc1019 std::unique_ptr<FunctionBody> host_fbody; in ConstructHostGraph() local
1023 fld, &host_fbody)); in ConstructHostGraph()
1028 FixupSourceAndSinkEdges(host_fbody->graph); in ConstructHostGraph()
1031 node_map[host_fbody->graph->source_node()] = (*host_graph)->source_node(); in ConstructHostGraph()
1032 node_map[host_fbody->graph->sink_node()] = (*host_graph)->sink_node(); in ConstructHostGraph()
1035 *host_fbody->graph, /*enter=*/nullptr, in ConstructHostGraph()