Searched refs:fdef_lib (Results 1 – 10 of 10) sorted by relevance
57 Status BuildXlaOps(const Scope& s, const FunctionDefLibrary& fdef_lib, in BuildXlaOps() argument61 FunctionLibraryDefinition flib_def(graph->op_registry(), fdef_lib); in BuildXlaOps()122 FunctionDefLibrary fdef_lib; in CreateFunctionDefLibWithConstFunction() local128 *fdef_lib.add_function() = std::move(func); in CreateFunctionDefLibWithConstFunction()129 return fdef_lib; in CreateFunctionDefLibWithConstFunction()136 FunctionDefLibrary fdef_lib = in TEST_F() local138 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()146 TF_ASSERT_OK(BuildXlaOps(root, fdef_lib, &graph)); in TEST_F()156 FunctionDefLibrary fdef_lib = in TEST_F() local158 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()[all …]
95 FunctionDefLibrary fdef_lib; in PartiallyDecluster() local96 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in PartiallyDecluster()
168 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() argument169 return graph_.AddFunctionLibrary(fdef_lib); in AddFunctionLibrary()
462 FunctionDefLibrary fdef_lib; in TEST_F() local463 *fdef_lib.add_function() = test::function::XTimesTwo(); in TEST_F()464 *fdef_lib.add_function() = test::function::XTimesFour(); in TEST_F()465 TF_ASSERT_OK(in_.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
602 Status Graph::AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() argument604 if (fdef_lib.function_size() > 0 && versions_->min_consumer() < 12) { in AddFunctionLibrary()607 return ops_.AddLibrary(fdef_lib); in AddFunctionLibrary()
532 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib);
655 tensorflow::FunctionDefLibrary fdef_lib; in TF_GraphCopyFunction() local656 *fdef_lib.add_function() = func->fdef; in TF_GraphCopyFunction()658 *fdef_lib.add_function() = grad->fdef; in TF_GraphCopyFunction()659 tensorflow::GradientDef* gdef = fdef_lib.add_gradient(); in TF_GraphCopyFunction()665 status->status = g->graph.AddFunctionLibrary(fdef_lib); in TF_GraphCopyFunction()
154 const auto& fdef_lib = gdef.library(); in CreateFunctionsFromTextProto() local155 if (fdef_lib.gradient_size() > 0) { in CreateFunctionsFromTextProto()162 for (const FunctionDef& fdef : fdef_lib.function()) { in CreateFunctionsFromTextProto()
340 FunctionDefLibrary fdef_lib; in AddNoinlineFunctionToGraph() local341 *(fdef_lib.add_function()) = GetNoinlineFunctionDef(); in AddNoinlineFunctionToGraph()342 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(fdef_lib)); in AddNoinlineFunctionToGraph()
565 FunctionDefLibrary fdef_lib; in TEST_F() local566 *(fdef_lib.add_function()) = fdef; in TEST_F()570 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()605 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST_F()