Home
last modified time | relevance | path

Searched refs:shape_inference_graphs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc245 std::vector<string> *shape_inference_graphs, in ExtractOutsideCompilationTest() argument
256 flr, fld, shape_inference_graphs, has_outside_compilation); in ExtractOutsideCompilationTest()
293 std::vector<string> shape_inference_graphs; in TEST_F() local
300 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
339 EXPECT_EQ(shape_inference_graphs.size(), 0); in TEST_F()
411 std::vector<string> shape_inference_graphs; in TEST_F() local
418 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
486 std::vector<string> shape_inference_graphs; in TEST_F() local
493 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
636 std::vector<string> shape_inference_graphs; in TEST_F() local
[all …]
Dextract_outside_compilation_pass.cc1811 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForFuncCallNode() argument
1840 shape_inference_graphs, &func_has_outside_compilation)); in ExtractOutsideCompilationForFuncCallNode()
1897 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForIfNode() argument
1916 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
1921 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
2016 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForWhileNode() argument
2033 fld, shape_inference_graphs, &cond_has_outside_compilation)); in ExtractOutsideCompilationForWhileNode()
2037 fld, shape_inference_graphs, &body_has_outside_compilation)); in ExtractOutsideCompilationForWhileNode()
2116 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions() argument
2132 host_compute_core, g, n, flr, fld, host_graphs, shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
[all …]
Dextract_outside_compilation_pass.h95 FunctionLibraryDefinition* fld, std::vector<string>* shape_inference_graphs,