Searched refs:reachable_flib (Results 1 – 2 of 2) sorted by relevance
1412 FunctionLibraryDefinition reachable_flib = flib.ReachableDefinitions(graph); in TEST() local1413 EXPECT_EQ(reachable_flib.num_functions(), 5); in TEST()1414 EXPECT_TRUE(reachable_flib.Contains("Func1")); in TEST()1415 EXPECT_TRUE(reachable_flib.Contains("Func2")); in TEST()1416 EXPECT_TRUE(reachable_flib.Contains("Func2_grad")); in TEST()1417 EXPECT_FALSE(reachable_flib.Contains("Func3")); in TEST()1418 EXPECT_TRUE(reachable_flib.Contains("Func4")); in TEST()1419 EXPECT_TRUE(reachable_flib.Contains("Func5")); in TEST()1420 EXPECT_FALSE(reachable_flib.Contains("Func6")); in TEST()
1460 FunctionLibraryDefinition reachable_flib(flib.default_registry(), in ReachableFunctionLibraryDefinition() local1468 const Status added = reachable_flib.AddFunctionDef(*func); in ReachableFunctionLibraryDefinition()1477 const Status added_grad = reachable_flib.AddGradientDef(grad); in ReachableFunctionLibraryDefinition()1482 return reachable_flib; in ReachableFunctionLibraryDefinition()