i0 :SSKJrJrJrJrJr SSKrSSKJs J s J s J r SSK r SSKJr SSKJr SSKJr SSKJrJr SSKJr SSKJr SS KJr "S S \ R<5r"S S \ R>5r"SS\ R@5r "SS\5r!g))AnyListOptionalSequenceTupleN)Tester) StageType)VulkanPartitioner)!get_symmetric_quantization_configVulkanQuantizer)EdgeCompileConfig) Partitioner) Quantizerc l^\rSrSrS S\\S\S-S\S\\\S\\4 U4Sjjjr S r U=r $) QuantizeN quantizerquantization_config calibratecalibration_samplesis_qatcl>[TU]U=(d [5U=(d [5UUUS9 g)N)rrrrr)super__init__r (get_symmetric_quantization_config_vulkan)selfrrrrr __class__s e/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/backends/vulkan/test/tester.pyrQuantize.__init__s7 4?#4#Q'O'Q 3  )NNTNF) __name__ __module__ __qualname____firstlineno__rrrboolrr__static_attributes__ __classcell__rs@rrrsd*.*.7;!&  I& !4Z   &hsm4     r rc:^\rSrSrSS\\4U4SjjjrSrU=r$) Partition- partitionercL>SS0n[TU]U=(d [U5S9 g)Nskip_bool_tensorsT)r-)rrr )rr-vk_compile_specrs rrPartition.__init__.s-.5 #I'8'I  r r!N) r"r#r$r%rrrr'r(r)s@rr+r+-s H[$9  r r+cN^\rSrSrSS\\\S\\4U4SjjjrSr U=r $)ToEdgeTransformAndLower5 partitionersedge_compile_configcr>UcSS0n[U5/n[TU] [UU=(d [SS9S9 g)Nr/TF)_check_ir_validity)default_partitioner_clsr6r7)r rrr )rr6r7r0rs rr ToEdgeTransformAndLower.__init__6sJ  2D9O-o>?L $5% 3!; E:  r r!)NN) r"r#r$r%rrrr rr'r(r)s@rr4r45s759;? tK01 &&78  r r4c ^\rSrSrSS\R R S\\RS\ \\ 4U4Sjjjr Sr U=r $) VulkanTesterGmoduleexample_inputsdynamic_shapesc>[RRRRR 5[ R[[ R[0-n[TU]1UUUUS9 g)N)r? stage_classesr@rA) executorchbackendstestharnessrdefault_stage_classesr PARTITIONr+TO_EDGE_TRANSFORM_AND_LOWERr4rr)rr?r@rArCrs rrVulkanTester.__init__Hso    $ $ , , 3 3 I I K##Y557N   '))  r r!r2)r"r#r$r%torchnnModulerTensorrrrr'r(r)s@rr=r=GsH 04   ell+ !s,   r r=)"typingrrrrrrD'executorch.backends.test.harness.stagesrErFrGstages BaseStagesrL executorch.backends.test.harnessr TesterBaser 9executorch.backends.vulkan.partitioner.vulkan_partitionerr 5executorch.backends.vulkan.quantizer.vulkan_quantizerr rr executorch.exirr #executorch.exir.backend.partitionerr#torchao.quantization.pt2e.quantizerrrr+r4r=r!r rr[st87<< A=W.;9 z"" ( $$  j@@ $ : r