Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Drearrange_function_argument_pass_test.cc54 FunctionDef *xla_fdef = fdl.add_function(); in TEST() local
55 TF_CHECK_OK(GraphToFunctionDef(*g, "f1", xla_fdef)); in TEST()
69 FunctionDef *xla_fdef = fdl.add_function(); in TEST() local
70 TF_CHECK_OK(GraphToFunctionDef(*g, "f2", xla_fdef)); in TEST()
82 FunctionDef *xla_fdef = fdl.add_function(); in TEST() local
83 TF_CHECK_OK(GraphToFunctionDef(*g, "f3", xla_fdef)); in TEST()
187 FunctionDef *xla_fdef = fdl.add_function(); in TEST() local
188 TF_CHECK_OK(GraphToFunctionDef(*g, "f2", xla_fdef)); in TEST()
202 FunctionDef *xla_fdef = fdl.add_function(); in TEST() local
203 TF_CHECK_OK(GraphToFunctionDef(*g, "f1", xla_fdef)); in TEST()
Dextract_outside_compilation_pass_test.cc286 FunctionDef *xla_fdef = fdl.add_function(); in TEST_F() local
287 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
404 FunctionDef *xla_fdef = fdl.add_function(); in TEST_F() local
405 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
479 FunctionDef *xla_fdef = fdl.add_function(); in TEST_F() local
480 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
629 FunctionDef *xla_fdef = fdl.add_function(); in TEST_F() local
630 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
765 FunctionDef *xla_fdef = fdl.add_function(); in TEST_F() local
766 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
[all …]
Dextract_outside_compilation_pass.cc2389 const FunctionDef* xla_fdef = fld->Find(n->name()); in ExtractOutsideCompilationForFunction() local
2390 if (!xla_fdef) { in ExtractOutsideCompilationForFunction()
2393 auto shape_inference_fdef = absl::make_unique<FunctionDef>(*xla_fdef); in ExtractOutsideCompilationForFunction()