Searched refs:reachable_flib (Results 1 – 2 of 2) sorted by relevance
1418 FunctionLibraryDefinition reachable_flib = flib.ReachableDefinitions(graph); in TEST() local1419 EXPECT_EQ(reachable_flib.num_functions(), 5); in TEST()1420 EXPECT_TRUE(reachable_flib.Contains("Func1")); in TEST()1421 EXPECT_TRUE(reachable_flib.Contains("Func2")); in TEST()1422 EXPECT_TRUE(reachable_flib.Contains("Func2_grad")); in TEST()1423 EXPECT_FALSE(reachable_flib.Contains("Func3")); in TEST()1424 EXPECT_TRUE(reachable_flib.Contains("Func4")); in TEST()1425 EXPECT_TRUE(reachable_flib.Contains("Func5")); in TEST()1426 EXPECT_FALSE(reachable_flib.Contains("Func6")); in TEST()
1705 FunctionLibraryDefinition reachable_flib(flib.default_registry(), in ReachableFunctionLibraryDefinition() local1711 Status added = reachable_flib.CopyFunctionDefFrom(func_name, flib); in ReachableFunctionLibraryDefinition()1720 const Status added_grad = reachable_flib.AddGradientDef(grad); in ReachableFunctionLibraryDefinition()1725 return reachable_flib; in ReachableFunctionLibraryDefinition()