EXPECT_EQ(fld.Find("host_graph"), nullptr);
}

TEST_F(ExtractOutsideCompilationForFunctionTest, OutsideCompilationInIf) {
  // Build the XLA computation func.
  // "const0" (bool)
  // "const1" (int32)
  // "if0" (pred = "const0", input = "const1", then_branch = "true_fn",
  //        else_branch = "false_fn")
  FunctionDefLibrary fdl;
  {
    tensorflow::Scope s = tensorflow::Scope::NewRootScope();

    %if0 = "tf.IfRegion"(%arg0) ({"tf.Yield"(%2) : (!tf_variant) -> ()},
                                 {"tf.Yield"(%2) : (!tf_variant) -> ()})
        {is_stateless = true} : (tensor<i1>) -> !tf_variant
    %if1 = "tf.IfRegion"(%arg0) ({"tf.Yield"(%if0) : (!tf_variant) -> ()},
                                 {"tf.Yield"(%if0) : (!tf_variant) -> ()})