i.SSKrSSKJrJr SSKJr SSKJrJr SSK J r J r J r SSK r SSKJrJr SSKJr SSKJr SS KJrJr SS KJrJr SS KJr SS KJr S SKJ r "SS\\5r!"SS\\!S9r""SS\#\5r$\"SS55r%\"SS55r&\"SS55r'\"S5\"SS555r(g)N)ABCMetaabstractmethod) dataclass)EnumEnumMeta)CallableListOptional)EdgeProgramManagerExportedProgram) experimental) Partitioner)EdgeCompileConfigExecutorchBackendConfig) PassManagerPassType) AOBaseConfig) Quantizer) StageTypec\rSrSrSrSrg)RecipeTypeMeta#z,Metaclass that combines EnumMeta and ABCMetaN)__name__ __module__ __qualname____firstlineno____doc____static_attributes__rW/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/export/recipe.pyrr#s6r!rc:\rSrSrSr\\S\4Sj55rSr g) RecipeType)z Base recipe type class that backends can extend to define their own recipe types. Backends should create their own enum classes that inherit from RecipeType: returncg)zr Return the backend name for this recipe type. Returns: str: The backend name (e.g., "xnnpack", "qnn", etc.) Nr)clss r"get_backend_nameRecipeType.get_backend_name/s r!rN) rrrrr classmethodrstrr)r rr!r"r$r$)s*    r!r$) metaclassc \rSrSrSrSrSrSrg)Mode;z Export mode enumeration. Attributes: DEBUG: Debug mode with additional checks and information RELEASE: Release mode optimized for performance debugreleaserN)rrrrrDEBUGRELEASEr rr!r"r/r/;s EGr!r/cn\rSrSr%Sr\\S'Sr\\ \ RR\ /\4\S'Srg)AOQuantizationConfigHz Configuration for torchao quantization with optional filter function. Attributes: ao_base_config: The AOBaseConfig for quantization filter_fn: Optional filter function to selectively apply quantization ao_base_configN filter_fnr)rrrrrr__annotations__r9r rtorchnnModuler,boolr rr!r"r6r6Hs5! BFIx%((//3!7!=>?Fr!r6ch\rSrSr%SrSr\\\\ S'Sr \\\ \ S'S\\\4Sjr Sr g) QuantizationRecipeVa] Configuration recipe for quantization. This class holds the configuration parameters for quantizing a model. Attributes: quantizers: Optional list of quantizers for model quantization ao_quantization_configs: Optional list of AOQuantizationConfig objects that pair AOBaseConfig with optional filter functions N quantizersao_quantization_configsr&cUR$)zm Get the quantizers associated with this recipe. Returns: The quantizers if any are set, otherwise None )rB)selfs r"get_quantizers!QuantizationRecipe.get_quantizersfsr!r)rrrrrrBr r rr:rCr6rFr rr!r"r@r@VsG -1Ji)0DHXd+?&@AHi 9r!r@c\rSrSr%SrSr\\\\ S'Sr S\\ \ \ /\\\-4-\ S'SrS\\ \/\\\-4-\ S'Sr\\\ S'Srg) LoweringRecipepa Configuration recipe for lowering and partitioning. This class holds the configuration parameters for lowering a model to backend-specific representations. Attributes: partitioners: Optional list of partitioners for model partitioning edge_transform_passes: Optional list of callables that take (method_name: str, exported_program: ExportedProgram) and return either List[PassType] or PassManager to be applied during edge lowering. edge_manager_transform_passes: Optional list of callables that take EdgeProgramManager as argument and return passes to be applied. Applied sequentially after TO_EDGE stage. edge_compile_config: Optional edge compilation configuration N partitionersedge_transform_passesedge_manager_transform_passesedge_compile_configr)rrrrrrKr r rr:rLrr,r rrrMr rNrr rr!r"rIrIps 15L(4 ,-4  tHc?3T(^k5QQRSS " tH014>K3OOPQQ 8<"34;r!rIzfThis API and all of its related functionality such as ExportSession and ExportRecipe are experimental.cT\rSrSr%SrSr\\\S'Sr \\ \S'Sr \\ \ \\/\4\S'Sr\\\S'Sr\\\S'Sr\\ \\S '\R,r\\S 'S r\\S '\SSj5r\SS\ SS\\S S4Sjj5r\SS\ SS\\S S4Sjj5rSrg) ExportRecipeab Configuration recipe for the export process. This class holds the configuration parameters for exporting a model, including compilation and transformation options. Attributes: name: Optional name for the recipe quantization_recipe: Optional quantization recipe for model quantization aten_transform_passes: Optional list of functions to apply transformation passes to the program before edge lowering. These callables are invoked to modify and return the transformed program. lowering_recipe: Optional lowering recipe for model lowering and partitioning executorch_backend_config: Optional backend configuration for ExecuTorch pipeline_stages: Optional list of stages to execute, defaults to a standard pipeline. mode: Export mode (debug or release) strict: Set the strict flag in the torch export call. Nnamequantization_recipeaten_transform_passeslowering_recipeexecutorch_backend_configpipeline_stagesmodeTstrictr&c @SSKJn [U[5(d[SU35eUR 5nUR "X40UD6nUcIUR U5n[SURSUSUVs/sHowRPM sn35eU$s snf)a Get an export recipe from backend. Backend is automatically determined based on the passed recipe type. Args: recipe: The type of recipe to create **kwargs: Recipe-specific parameters Returns: ExportRecipe configured for the specified recipe type r)recipe_registryzInvalid recipe type: zRecipe 'z' not supported by 'z'. Supported: )r[ isinstancer$ ValueErrorr) create_recipeget_supported_recipesvalue)r(recipekwargsr[backend export_recipe supportedrs r" get_recipeExportRecipe.get_recipes 5&*--4VH=> >))+'55fPP  '==gFI6<<.((>(K(KL%%&*@*@*V*V$++**@@ '-&@&@@"@&11"))*=*H*HI'>>/66+CC '.63S3S3S*.--44+';&D(,$ 8+=-;>3N/TX, ($( 3"& )***..BBB*0*@*@*T*T' *(61A-t,@(d$7$N;L;N ( $" SXX, C_VQVV_ C&  <"54&=    Ds J2Jr)rar$r&rP)N)rrrrrrRr r,r:rSr@rTr rr rUrIrVrrWrr/r4rXrYr>r+rgrnrmr rr!r"rPrPs) $D(3-8<"45< 8 XsO,o= >? 15OXn-4CGx(?@G15OXd9o.5D$FD6IM:>*:9A#: :::QU] ">2] AI#] ] ] r!rP))rtabcrr dataclassesrenumrrtypingrr r r;executorch.exirr r executorch.exir._warningsr #executorch.exir.backend.partitionerrexecutorch.exir.capturerrexecutorch.exir.pass_managerrrtorchao.core.configr#torchao.quantization.pt2e.quantizerrtypesrrr$r,r/r6r@rIrPrr!r"rs '!++ ?2;N>,9 Xw   $ 3   G G  G  2 << <8l w w  w r!