Searched refs:optimized_flib (Results 1 – 3 of 3) sorted by relevance
303 FunctionLibraryDefinition optimized_flib(OpRegistry::Global(), in TEST_F() local307 EXPECT_EQ(5, optimized_flib.num_functions()); in TEST_F()330 const FunctionDef* optimized_func_0 = optimized_flib.Find(optimized_0); in TEST_F()331 const FunctionDef* optimized_func_1 = optimized_flib.Find(optimized_1); in TEST_F()332 const FunctionDef* optimized_func_2 = optimized_flib.Find(optimized_2); in TEST_F()454 FunctionLibraryDefinition optimized_flib(OpRegistry::Global(), in TEST_F() local458 EXPECT_EQ(3, optimized_flib.num_functions()); in TEST_F()479 const FunctionDef* my_mul_spec = optimized_flib.Find(specialized_my_mul); in TEST_F()480 const FunctionDef* my_fwd_spec = optimized_flib.Find(specialized_my_fwd); in TEST_F()550 FunctionLibraryDefinition optimized_flib(OpRegistry::Global(), in TEST_F() local[all …]
597 std::unique_ptr<FunctionLibraryDefinition>* optimized_flib) { in OptimizeGraph() argument700 optimized_flib->reset(new FunctionLibraryDefinition(*flib_def_)); in OptimizeGraph()705 if ((*optimized_flib)->Contains(func_name)) { in OptimizeGraph()707 TF_RETURN_IF_ERROR((*optimized_flib)->ReplaceFunction(func_name, fdef)); in OptimizeGraph()710 TF_RETURN_IF_ERROR((*optimized_flib)->AddFunctionDef(fdef)); in OptimizeGraph()750 std::unique_ptr<FunctionLibraryDefinition> optimized_flib; in BuildGraph() local752 Status s = OptimizeGraph(options, &optimized_graph, &optimized_flib); in BuildGraph()759 optimized_flib.reset(new FunctionLibraryDefinition(*flib_def_)); in BuildGraph()784 optimization_options.flib_def = optimized_flib.get(); in BuildGraph()804 const FunctionDef* fdef = optimized_flib->Find(node->def().op()); in BuildGraph()[all …]
192 std::unique_ptr<FunctionLibraryDefinition>* optimized_flib);