Searched refs:shape_inference_graphs (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass_test.cc | 245 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 …]
|
| H A D | extract_outside_compilation_pass.cc | 1814 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForFuncCallNode() argument
1843 shape_inference_graphs, &func_has_outside_compilation)); in ExtractOutsideCompilationForFuncCallNode()
1900 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForIfNode() argument
1919 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
1924 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
2019 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForWhileNode() argument
2036 fld, shape_inference_graphs, &cond_has_outside_compilation)); in ExtractOutsideCompilationForWhileNode()
2040 fld, shape_inference_graphs, &body_has_outside_compilation)); in ExtractOutsideCompilationForWhileNode()
2119 std::vector<string>* shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions() argument
2135 host_compute_core, g, n, flr, fld, host_graphs, shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
[all …]
|
| H A D | extract_outside_compilation_pass.h | 95 FunctionLibraryDefinition* fld, std::vector<string>* shape_inference_graphs,
|