Wheelodex » Projects » ai-edge-torch-nightly

ai-edge-torch-nightly

View on PyPI — Reverse Dependencies (0)

0.3.0.dev20241121 ai_edge_torch_nightly-0.3.0.dev20241121-py3-none-any.whl
0.3.0.dev20241120 ai_edge_torch_nightly-0.3.0.dev20241120-py3-none-any.whl

Wheel Details

Project: ai-edge-torch-nightly
Version: 0.3.0.dev20241120
Filename: ai_edge_torch_nightly-0.3.0.dev20241120-py3-none-any.whl
Download: [link]
Size: 316207
MD5: 2a71c8435b159378af78f6dd103c9b99
SHA256: 8b926d42c1f0cd09e9f83ee63d9c0013033c8e70c53ca249100f20ae843897c5
Uploaded: 2024-11-20 20:22:25 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: ai-edge-torch-nightly
Version: 0.3.0.dev20241120
Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
Home-Page: https://github.com/google-ai-edge/ai-edge-torch
Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.9
Requires-Dist: numpy
Requires-Dist: scipy
Requires-Dist: safetensors
Requires-Dist: tabulate
Requires-Dist: torch (>=2.4.0)
Requires-Dist: torch-xla (>=2.4.0)
Requires-Dist: tf-nightly (>=2.19.0.dev20241001)
Requires-Dist: ai-edge-litert-nightly
Requires-Dist: ai-edge-quantizer-nightly
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 420 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.45.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
ai_edge_torch/__init__.py sha256=48qP37uHT90YPs4eIUQxCiWVwqGEX3idCUs6mQKvX1U 1168
ai_edge_torch/config.py sha256=FMWeCH2b7HYILBvaI1iZNnYCO4WAhDOwBZBmIE-xrF0 909
ai_edge_torch/conftest.py sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI 758
ai_edge_torch/fx_pass_base.py sha256=518ziQ0TUxqum2qZXqlD8qr65pHPh8ZNLnwFC6zvK3k 4253
ai_edge_torch/model.py sha256=N-pNpTxzhaFGhWhnSGd70lBzb9VlEhTOq5mddU7bvvI 5542
ai_edge_torch/version.py sha256=52sF7t2CBQE8RcB2Hcmo-f6_BLyCW9NzWZ-wTKM9ho4 706
ai_edge_torch/_convert/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/_convert/conversion.py sha256=HwzfRx_DX5TLtPqwEH1_NOm38_INvHzHl4_mX67KOdQ 5448
ai_edge_torch/_convert/conversion_utils.py sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o 2160
ai_edge_torch/_convert/converter.py sha256=DYbTZMZos8bvm9mLyDv3W1P8ER_iGKVohbFAmLZD4r8 9534
ai_edge_torch/_convert/signature.py sha256=rGpBNss3Y9FCCCcdBwDo16KqavJi8N5P0M_6Wnmoz6Y 2320
ai_edge_torch/_convert/to_channel_last_io.py sha256=_31phf7TYgZY2ftpNbrdlB1RhDium1lz_BXEQ6IsMFc 2893
ai_edge_torch/_convert/fx_passes/__init__.py sha256=NVe-eGcm7j8jZpP2pcMhC8j5dVjgR1pPzyXhHdvKH4E 1267
ai_edge_torch/_convert/fx_passes/build_aten_composite_pass.py sha256=doaww8KqrgRTD5LotBVAIRFsEqzPn9R5lcGehBJOczA 9098
ai_edge_torch/_convert/fx_passes/build_interpolate_composite_pass.py sha256=qb4JBDi4Xca14JJUIcaaZQIJiyqKyHJF49jsRCIFCVA 4335
ai_edge_torch/_convert/fx_passes/inject_mlir_debuginfo_pass.py sha256=IlZuK42kfVcRqAWZp4j2k_81T2uWo9T2558U_GPJAlU 2327
ai_edge_torch/_convert/fx_passes/remove_non_user_outputs_pass.py sha256=f1IUVWyhioOClsMiZzLyynoW2R17U83vA-7Q-3pGPM4 2126
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/__init__.py sha256=lxnoH-WGLeiQIF8XjMGodjiZEFTxucl7g05N7MR9OPk 796
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_check.py sha256=hDsl9AHzmyuSWsdHOSO114l4nBUgUdAOUWafMTipMgA 7629
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_mark.py sha256=4RyGUwR22bZqkn_TnptenFJodc_Q43f4_SBG7gmTbos 1621
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_rewrite.py sha256=NW37V6QYdPOZOVhqLcmssVk-VAeO4ECk_CrbEBh4B0E 12740
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/op_func_registry.py sha256=bsYnudRlXp1PJlu4GF25KSogSkBGQPSaecBrUTONKaw 1031
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/pass_body.py sha256=t94Am3iPbYQekg-rrtc-jS_aDWtEgAAj7pAKHrG0-9U 10563
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/utils.py sha256=YLMttMg5PdvXTtQ8lxpKb434UGVvYVALV1-xeuH4UGc 2131
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/__init__.py sha256=D8VX8SbCzfoyvPgMFHK7yxD7R-bzLxp2gfdKxgrWekA 742
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/greedy.py sha256=L_x8BrF7UDah-SYl-pG11I6CIckdU9kBTUHcmwW4cts 2420
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/min_cut.py sha256=mzfL9cf0qBnpmxM_OlMQFvQsEZV2B_Mia9yEJV4J7rI 7135
ai_edge_torch/_convert/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/_convert/test/test_convert.py sha256=yXfeWDw9u_rTS3B6kvvFPo5E4XNT3zKTSLFSBSAI9Fc 15502
ai_edge_torch/_convert/test/test_convert_composites.py sha256=ELwHxTdTTCJm30aWg_PZXxg9HvDM4Hnf9lT0wwOWT6s 8060
ai_edge_torch/_convert/test/test_convert_multisig.py sha256=6_C2R9--KyNR7_oezZIAfyTSR97tOeEWy4XGcbSxBDE 5778
ai_edge_torch/_convert/test/test_to_channel_last_io.py sha256=1o-gUiwzIuO67FNAJ8DeyKv8fVUeZVNNNwofNVDjYeU 3024
ai_edge_torch/debug/__init__.py sha256=N05Mmvi41KgSuK0JhuMejERESgP8QekiGdp9_PEyuKU 742
ai_edge_torch/debug/culprit.py sha256=7UYVpVWpiCXbMAyThVtHt_kc_poT7sCTh5UUPvcycgk 14832
ai_edge_torch/debug/utils.py sha256=vOAL4t6Lj47uhKapfEsc_WHmvwew3eKO9hSJyzvPXnU 1625
ai_edge_torch/debug/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/debug/test/test_culprit.py sha256=SLX4rC-5Dlna8MWHhGRNe72K71AHTFufDrWLlFQn50c 3773
ai_edge_torch/debug/test/test_search_model.py sha256=-RuU0QsjqkfzZF2IbeA55MoeVOawhbgiSEu96PmioPE 1668
ai_edge_torch/experimental/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/amd_llama_135m/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/amd_llama_135m/amd_llama_135m.py sha256=bkq2ZknJfuY7WC8wLVg92Z6eA_aMDbkgwaMxvmDW4_0 2618
ai_edge_torch/generative/examples/amd_llama_135m/convert_to_tflite.py sha256=-n79r6yFnCACpms5eMkXNpyQsCn2PYVRdB-jOoIqn14 2227
ai_edge_torch/generative/examples/amd_llama_135m/verify.py sha256=-9Nb9D818YSJR3olVtBwoLNeMMD5qE58YBnsA67hlHg 2421
ai_edge_torch/generative/examples/gemma/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/gemma/convert_gemma1_to_tflite.py sha256=evmUj_4yygQthSRU-ke-Xn1qFNDCZKbegqINWfruKwU 2184
ai_edge_torch/generative/examples/gemma/convert_gemma2_multi_prefills.py sha256=6d9wG5MnStEys34_gFXwKTMRXUBFLTW1jEzCoWkAtwM 2224
ai_edge_torch/generative/examples/gemma/convert_gemma2_to_tflite.py sha256=RZDs6oY-NLYrPNtfuJDweIHzGUL2kzpIc3AW_1p8gGg 2186
ai_edge_torch/generative/examples/gemma/gemma1.py sha256=oSbysiPvwp5efMbNYZop3HrxDMGiD15Tmz-HiQuTr2E 3315
ai_edge_torch/generative/examples/gemma/gemma2.py sha256=RQFQDMEnIVp8PefcCTr7P0CvllKI7FVoIJLXbPLLIsc 9056
ai_edge_torch/generative/examples/gemma/verify_gemma1.py sha256=ip-Gmk4CI5f0GWSdAIdrectxQWJ0t328KCsA4nfHuGg 1736
ai_edge_torch/generative/examples/gemma/verify_gemma2.py sha256=IoBhEMwH07-tFm5-U6F2hpCsI8xynglhq1x9tIOdaPQ 1322
ai_edge_torch/generative/examples/gemma/verify_util.py sha256=tR8RflXocDZqvuStyw9aFlzuiTllEC8rNnjrxms6_Is 5727
ai_edge_torch/generative/examples/llama/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/llama/convert_to_tflite.py sha256=P0-pByTM5tslE23ILgo7nd0nOGE25ciBRG5wKJj0bBk 2411
ai_edge_torch/generative/examples/llama/llama.py sha256=AMcCbuDBxEfbO-l3KiEXbUaXEJ3RLLwkHii7to7UhVo 6854
ai_edge_torch/generative/examples/llama/verify.py sha256=X7oKQi85M789ugBrOlMvzk8eSRR3Kf1Mprfl-U-WIpo 2842
ai_edge_torch/generative/examples/openelm/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/openelm/convert_to_tflite.py sha256=85FVEt6cKFP2UzCLC78tAkbwGlGhAArtG7Wa75NxJik 2185
ai_edge_torch/generative/examples/openelm/openelm.py sha256=sFakstoPDcOHSak0IGFEEq_HQMBBSMcx-WVCDZqcVDo 4411
ai_edge_torch/generative/examples/openelm/verify.py sha256=VkigoqhAr8ew95neb3TifYv-SLOSheaWKv2AH0iKDrc 2441
ai_edge_torch/generative/examples/paligemma/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/paligemma/convert_to_tflite.py sha256=dT7dnx1dzGzFiH5gQJ4M6zcTLSRFvSDpi3IuZ9_vd78 2706
ai_edge_torch/generative/examples/paligemma/decoder.py sha256=XMeznGBbjRJidv725L6_7XzkYskS2cDjf8NGB18FNhg 4944
ai_edge_torch/generative/examples/paligemma/image_encoder.py sha256=yKPWG8aBp-GuzeyQntlzwTTcGBBjvUywVGRjnlNprmo 5574
ai_edge_torch/generative/examples/paligemma/paligemma.py sha256=pIjsS-IUFevRjFA9153YT1vtWXATGWHsgVQQX_nWaZQ 5280
ai_edge_torch/generative/examples/paligemma/verify.py sha256=Bkbgy-GFjnMNYjduWUM7YLWarPTwmj1v38eHY-PdBlM 4874
ai_edge_torch/generative/examples/paligemma/verify_decoder.py sha256=al5wMPWri4IRVWrLmCplPi6uoCzwh0vBHMGnCt-XUqo 2690
ai_edge_torch/generative/examples/paligemma/verify_image_encoder.py sha256=pSekf1BybhieQz3cQx_llbRQHxczXbTqool8fOyGj_0 3114
ai_edge_torch/generative/examples/phi/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/phi/convert_phi3_to_tflite.py sha256=rkbTtMaqSVG48cm-NTxR_LDgZmXAEBqayTm9O49oMXc 2171
ai_edge_torch/generative/examples/phi/convert_to_tflite.py sha256=3go690yX6PFeXMdpY7y4JZorAwxX0HT_b_pKZieauvk 2169
ai_edge_torch/generative/examples/phi/phi2.py sha256=nbivDwZREd-sypy_ittO59-yaAdPvHv1YEV6Fo5buCo 3341
ai_edge_torch/generative/examples/phi/phi3.py sha256=GkHOaYfsFEbHvfZCaLlb3Us_h19ezqPDUakoz_DiG9A 7123
ai_edge_torch/generative/examples/phi/verify.py sha256=YPFCdbnfmvq38fbpBNr0kHPfSZo4p3_6WkLJAW3pLPo 2177
ai_edge_torch/generative/examples/phi/verify_phi3.py sha256=kVYaBVvddfQng0IyZGxyTJEzhiPO0G4VFJm2WOc2Q94 2360
ai_edge_torch/generative/examples/qwen/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/qwen/convert_to_tflite.py sha256=QAAVoSKDVf2rHAChzumGloVCWIU0Oe5UYKgv3T192Iw 2496
ai_edge_torch/generative/examples/qwen/qwen.py sha256=oYm9hhALUQ4uOn-PO1bF7fCIGP8EWRNK4zClkx2RQs8 4070
ai_edge_torch/generative/examples/qwen/verify.py sha256=9_AyEJTeUfvhhID64Rto2bflFPyXMFokdQLsseLUMiI 2775
ai_edge_torch/generative/examples/smollm/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/smollm/convert_to_tflite.py sha256=zPrDTDeRVWFi9DS32uNi-RLpzOStFOk5MhNla4ixeew 2179
ai_edge_torch/generative/examples/smollm/smollm.py sha256=M5qAcSUE5gxOSfq24a8lZku9kgvmlFCyIBar3kF2XEk 2570
ai_edge_torch/generative/examples/smollm/verify.py sha256=HXYcCjDJMylVL3Pc9HU-UXqtpjtIU25o1YhPiX30aPU 2361
ai_edge_torch/generative/examples/stable_diffusion/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/stable_diffusion/attention.py sha256=kDWG6MlIGa89zC5KSRcJlw2c4ITuw8KcchtfmF55f4g 3545
ai_edge_torch/generative/examples/stable_diffusion/clip.py sha256=5M4auM33SgCTODt0VT8TO-EVILruqGDRiNILBPeB83Y 6072
ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py sha256=sB_7-PVri8PxKnFG7c8GsTGyrxGEda-oZwGyyScTL3o 5239
ai_edge_torch/generative/examples/stable_diffusion/decoder.py sha256=sQKQ-k6H9kG2brgwLsktjCMeN2h0POyfMP6iNsPNKWc 16271
ai_edge_torch/generative/examples/stable_diffusion/diffusion.py sha256=6W58LxmHHkz2ctgpknQkyoDANZAnE9Byp_svfqLpQf0 34793
ai_edge_torch/generative/examples/stable_diffusion/encoder.py sha256=CAPsW84A8f00nS6fLFeh_XUjCPsDCA5UxHOUsMrLfSU 3450
ai_edge_torch/generative/examples/stable_diffusion/pipeline.py sha256=GnY3vPZ-obrWuJifuE5bUooKLqAI7v6q71oaTuLKeBE 8778
ai_edge_torch/generative/examples/stable_diffusion/tokenizer.py sha256=xychak9hdLd6ieXBYEwrK2BkF8NRZWZSSCijIsESpBA 3420
ai_edge_torch/generative/examples/stable_diffusion/util.py sha256=XIXIB0vCvQKOGyIyiZeiIA5DLeSXjkudywvJS4FK7AM 2431
ai_edge_torch/generative/examples/stable_diffusion/samplers/__init__.py sha256=uQWKzCD_49ackNFrt50H04dkDXxfAwUCtMWWQre5SVE 830
ai_edge_torch/generative/examples/stable_diffusion/samplers/k_euler.py sha256=wBBNM24waZ57M1rXonwesfUkKe9DqpqO3eW6BfZkrD0 2323
ai_edge_torch/generative/examples/stable_diffusion/samplers/k_euler_ancestral.py sha256=c89ldwtuQ2_yspGrGa7oh7fsvTt6A86Whxa6fBK9YOQ 2526
ai_edge_torch/generative/examples/stable_diffusion/samplers/k_lms.py sha256=ZE6HyOoBJrmTh54KVFf7DjNBnBS0pT4cgviYaq8HGMU 2801
ai_edge_torch/generative/examples/stable_diffusion/samplers/sampler.py sha256=RxR5rw0wFFm_5CfAY-3-EIz83vhM9EKye8Bb5zBb0Ok 1341
ai_edge_torch/generative/examples/t5/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/t5/convert_to_tflite.py sha256=HHtZTtUh3QgE4F74-ru_8n1pt6cqfbObw12xoaMJ7NQ 4596
ai_edge_torch/generative/examples/t5/t5.py sha256=gFTmPi-xB8pcPRgoF3DJxvH_fT-KWTb8ii77P5UbKR0 21263
ai_edge_torch/generative/examples/t5/t5_attention.py sha256=l01oYyJo77INzRwN4xqXquaFQPvCFBFF5zOnmGVb3Hg 8731
ai_edge_torch/generative/examples/test_models/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/test_models/convert_toy_model.py sha256=6-WaNHckq_LlXMVTh8x90MGWeWq2bu_T_XQd3w9FnGg 3261
ai_edge_torch/generative/examples/test_models/toy_model.py sha256=4113jZK-Hu3kYop__WTc8Bq-bG6YzQtADbxHtYPEB4w 5036
ai_edge_torch/generative/examples/test_models/toy_model_with_kv_cache.py sha256=rRodLr-hEqAs_-8x06O8qO-hJ_cqr2AfhJZ9DCptvwo 4332
ai_edge_torch/generative/examples/tiny_llama/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/tiny_llama/convert_to_tflite.py sha256=ekxd8efjMgEvauUu3PidWOC-DszPHn5sqU753F7sJIM 2201
ai_edge_torch/generative/examples/tiny_llama/tiny_llama.py sha256=10X8HwPx4akzclnIMOBNItKQemhRbvxBbTo7nwZtWjM 2650
ai_edge_torch/generative/examples/tiny_llama/verify.py sha256=7Bk8z033M-BCXJ299fpQNXYAudBbZoDQp9934xcvg50 2426
ai_edge_torch/generative/fx_passes/__init__.py sha256=jrzCB3ZyY_t5jJM1e2Czdt3DjAIL43R0_a-T-I7wOzw 1155
ai_edge_torch/generative/fx_passes/remove_sdpa_zero_mask_pass.py sha256=hhxSQvkDMv0isZJhmuLiod66ZODaJ8uSPSVTJVHBabQ 1931
ai_edge_torch/generative/layers/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/layers/attention.py sha256=zN3BQjA25Ej_aRU0rFnyx--K74xf5ykc02zGvUpYHeE 13295
ai_edge_torch/generative/layers/attention_utils.py sha256=zBVwlBUTs-nStIKCZG0ks5ra7tsqc9ShfakFJKH5rds 7344
ai_edge_torch/generative/layers/builder.py sha256=Z5LyzCEThgnYZeyViakaE3yJVzTGHtw13acHsAQR15U 5050
ai_edge_torch/generative/layers/feed_forward.py sha256=hdICat-8gW7-vxDAevJQ8NQ-mynllPiqLdXQMF6JMnc 4189
ai_edge_torch/generative/layers/kv_cache.py sha256=lbm-yJ1jGPtcgWS4C3FmSnB1IlxqDE7g0BLRh3PN4N4 6324
ai_edge_torch/generative/layers/model_config.py sha256=viX51T_naJ9sPpPxPoMnSueBPYE2zxWNOD0xn0f-_bM 7510
ai_edge_torch/generative/layers/normalization.py sha256=eKAGst9rPuyRFExMcQFJO7R3iHdCtlmjeF_lITjLhwE 6498
ai_edge_torch/generative/layers/rotary_position_embedding.py sha256=CZqOoibLcHvUgrgaIIWAlmk3XgE2inzx340MN-npLoU 1347
ai_edge_torch/generative/layers/scaled_dot_product_attention.py sha256=gXxh3papKy4FBpGEX7VyZ7rZ1Js6aHK70Q6DKrVSckY 4154
ai_edge_torch/generative/layers/unet/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/layers/unet/blocks_2d.py sha256=9jKzOfiBQ66bp1ZnVIAoREIifVNFx4aTlQeYMAx2_pA 29062
ai_edge_torch/generative/layers/unet/builder.py sha256=zAqWXdimmMrQRhmE_t9XkS68mh6PSrzwb-2NZZXrR5I 1901
ai_edge_torch/generative/layers/unet/model_config.py sha256=pPDwLawc23pfMaPVyMJlYmxVVusjMvx-l8wBwOYOH-c 9692
ai_edge_torch/generative/quantize/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/quantize/example.py sha256=1lfVNUd2cEyRUnoZ7BLbRJ9IN-FTKiWBtZNPFUzAiWE 1747
ai_edge_torch/generative/quantize/quant_attrs.py sha256=n1Fm8BFC8gJa_oiwwAOOghJyHtOXYZ4q-5ZRy4pHrIw 1957
ai_edge_torch/generative/quantize/quant_recipe.py sha256=tKnuJq6hPD23JPCB9nPAlE1UHAwdbChkgPShiVaz4CE 5156
ai_edge_torch/generative/quantize/quant_recipe_utils.py sha256=4fgmP_GgeiFUOkIaC9ZZXC12eO3DQZdrWDXRz5YXiwU 2270
ai_edge_torch/generative/quantize/quant_recipes.py sha256=0Kvr_o7pbMnE8VMe6Ml0FBxkHM6RJ3C14B2I1mjItjc 2030
ai_edge_torch/generative/quantize/supported_schemes.py sha256=FjdycEOvxRgBmQdZVufetPvkDoD7rUowIOSKV9oV5Kk 1418
ai_edge_torch/generative/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/test/test_kv_cache.py sha256=W6Bh0gYDzmwb0j9HdD5_D7Z7FPToP2HSyFrmwIXuFqo 3793
ai_edge_torch/generative/test/test_loader.py sha256=9mQUeeZKOVApOWSWl2cN9c10axZjMKM1-0Zd823CCS4 3449
ai_edge_torch/generative/test/test_model_conversion.py sha256=aZFaheg2sq7rEccch1TZM6W4BSfpJZjrM9Gyp4hVGYs 6351
ai_edge_torch/generative/test/test_model_conversion_large.py sha256=xWV9O2wuRHc4VNBWuWipiuqXa3AJhiV1nmjewAZHHWM 11177
ai_edge_torch/generative/test/test_quantize.py sha256=8geJhKwYBU20m0mdGPD1BUFwQ0lZKNtCB04SOLO18y4 5980
ai_edge_torch/generative/test/utils.py sha256=eQ-hjd1eXuHJF3SJK6_CrjgOZVzmG_4VEdH7Z1gH_lA 1897
ai_edge_torch/generative/utilities/__init__.py sha256=-_jxnnFnCgnTU4oTm4MnRsvL5lqhomBNdFBbqfmfHPo 720
ai_edge_torch/generative/utilities/converter.py sha256=S14STbyxV6A9HKy1BdUo49f2jS6Ij0RL9mVAFUMWYV8 5291
ai_edge_torch/generative/utilities/loader.py sha256=A3SOjPXp--AsvoP1hqj5QKWE4sgxoFc3H5EBUz_Eogc 13531
ai_edge_torch/generative/utilities/model_builder.py sha256=OcHJhEqc3LjI3STli6cyn71m1mdzr7QbzF9fqSNCXrg 5730
ai_edge_torch/generative/utilities/stable_diffusion_loader.py sha256=dqPD9qRXEWtU3ombslOC-BE2l_dMwHoCNu7NsIJhsso 36158
ai_edge_torch/generative/utilities/t5_loader.py sha256=tEsfy8-ymzbbjOIc-oesXF3yGyyWtJgFXn2s7VOavt8 16961
ai_edge_torch/generative/utilities/transformers_verifier.py sha256=8sp9m_FMcXn7nqOrochtu2jIANkJKhnhIBUmH0ZTDR4 1549
ai_edge_torch/generative/utilities/verifier.py sha256=GLh7h8pcpSKtCKoPyxJhv3TmvENd2h6ek_cnbe2s3Ak 11418
ai_edge_torch/hlfb/__init__.py sha256=sH4um75na-O8tzxN6chFyp6Y4xnexsE7kUQpZySv6dE 735
ai_edge_torch/hlfb/mark_pattern/__init__.py sha256=cjTprggj_cuktSCm7-A25e7Shop3k63ylp7sdZmtZ8o 4790
ai_edge_torch/hlfb/mark_pattern/passes.py sha256=pjkKcI1nHECPluAt87cFBrt1DP0f3ge7rHq1NhCkBIE 1936
ai_edge_torch/hlfb/mark_pattern/pattern.py sha256=NP2mYhe5D2GjtqQfqqldp-ko3xtNghuFKKJOQskUJFI 10041
ai_edge_torch/hlfb/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/hlfb/test/test_mark_pattern.py sha256=ivq0eVjuf31idfNY0E12F4FxdkSI9hwYXapLJBkIf8Q 4831
ai_edge_torch/hlfb/test/test_stablehlo_composite_builder.py sha256=j8WpeS-mz3Zr4I7p7NwanQzkQNeH0asZ7lz5y7twgQ4 8447
ai_edge_torch/lowertools/__init__.py sha256=A8WBXvWtuFYYWtNTqPD7waVntLaSVAnSMwx5ugjZBIw 761
ai_edge_torch/lowertools/_shim.py sha256=ilL7x1ebUBj1clg7bagrX4y_nVSHiGrvDrOVfuTeenE 3039
ai_edge_torch/lowertools/common_utils.py sha256=Z7p-ivOHtddktpnHrlDm_dSoTxJOdEjFXIGQbzjgwQo 4504
ai_edge_torch/lowertools/odml_torch_utils.py sha256=K5dZ_fFDL3GWKo0IoY4OC_GX5MY-guY-MqteolyV9hg 8098
ai_edge_torch/lowertools/test_utils.py sha256=bPgc2iXX16KYtMNvmsRdKfrCY6UJmcfitfCOvHoD7Oc 1930
ai_edge_torch/lowertools/torch_xla_utils.py sha256=XGZE0vZG9WSQT-6dFmPlU8W89z8rfXPRGjuZeuhXCIw 9205
ai_edge_torch/lowertools/translate_recipe.py sha256=ymkBpFqAUiupRWqrPOWiVphKcXR1K5vHK0RjgBFtxlE 5652
ai_edge_torch/odml_torch/__init__.py sha256=S8jOzE9nLof-6es3XDiGJRN-9H_XTxsVm9dE7lD3RWo 812
ai_edge_torch/odml_torch/_torch_future.py sha256=AJ0klpsbu2ZBTfiZlqSOoaYzBVITt40a1fYN8xKkEPw 3044
ai_edge_torch/odml_torch/_torch_library.py sha256=Lw1gqL2HWNRspdTwNhIkYAHDyafHedHtkXyKKxn-Wss 805
ai_edge_torch/odml_torch/export.py sha256=4xwrsDeOAgzoB9m7EeNsBj6dC5Ajtn5aKDRQkdHxa-o 11584
ai_edge_torch/odml_torch/export_utils.py sha256=q84U69ZQ82hLXw-xncJ8IW-K71Xux-NWlzZTs7hdZWA 5127
ai_edge_torch/odml_torch/tf_integration.py sha256=lTFJPPEijLPFmn6qq2jbpVTQOo0YaOTK36kK6rCiyIE 5956
ai_edge_torch/odml_torch/composite/__init__.py sha256=71GM_gDZxJyo38ZSoYSwhZX3xKA9rknO93JS9kw9w_c 778
ai_edge_torch/odml_torch/composite/mark_tensor.py sha256=U--rwl-XkWKgkdXCXDn6yySug8FR66o1YFUAIoSaWW4 3523
ai_edge_torch/odml_torch/composite/stablehlo_composite_builder.py sha256=2Y52E_gLeoXpMcPpV-svXsgN3JbEIjnPVjm0xkpTUdQ 3319
ai_edge_torch/odml_torch/debuginfo/__init__.py sha256=9ag6-WWRG50rPCtIV7OpIokEKu2YRyGlMZZqVPWUH6g 762
ai_edge_torch/odml_torch/debuginfo/_build.py sha256=1xCXOs3-9UcsOyLFH0uyQwLu7c06iYFTo0NQ7Ckbl2I 1465
ai_edge_torch/odml_torch/debuginfo/_op_polyfill.py sha256=IvOBQyROI9WHS3umHRxsDW-1YElU9BPWzKtJA2eKWOI 1739
ai_edge_torch/odml_torch/jax_bridge/__init__.py sha256=Jco5zvejxuyl9xHQxZICAKbkgH7x38qPlwUUpD7S15Q 730
ai_edge_torch/odml_torch/jax_bridge/_wrap.py sha256=drN3L0uTsSjkluKgt6Ngq7b5HLReE_7iAitHpZ9PKqE 5428
ai_edge_torch/odml_torch/jax_bridge/utils.py sha256=T8isGc896VrHZ6c_L5pYmLpolQ7ibcOlgWfPuVFPzIg 2264
ai_edge_torch/odml_torch/lowerings/__init__.py sha256=dE_qzh-OnCNjWzqs1-PHs5PNlRF726qMQKM3tkwAzEs 959
ai_edge_torch/odml_torch/lowerings/_basic.py sha256=z_hPJX9n97d6obcsS9OHXpKqbmw6QqACXgnq5ML6Rhs 9014
ai_edge_torch/odml_torch/lowerings/_batch_norm.py sha256=PaLI0BB6pdBW1VyfW8VTOT_Be-ZcqYdNOsyfzKfq8Cg 2064
ai_edge_torch/odml_torch/lowerings/_convolution.py sha256=v1VdKmL8YLJv3PR9VgyNghO83A25PpTzY2ZUAJqlq3Q 6847
ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py sha256=1JeX3j7Rt3KE7Z2eYRrhtcYgO3EKnRyZFKAUWXw-bsU 10812
ai_edge_torch/odml_torch/lowerings/_layer_norm.py sha256=1ePJs7oIdUkVdMddFsXMc53qTkEKqGz0ZhQQoNzBa10 2862
ai_edge_torch/odml_torch/lowerings/context.py sha256=jslcCv7r_HtImSRTxJwHAUV_QCu9Jub51lovmoBkmFA 1295
ai_edge_torch/odml_torch/lowerings/registry.py sha256=gqx3n1Mx8pnGQz3nkIF1T_8bkRabXLJBvUoJJn5kOUY 2911
ai_edge_torch/odml_torch/lowerings/utils.py sha256=NczqpsSd3Fn7yVcPC3qllemiZxxDAZgcW1T5l8-W9fE 5593
ai_edge_torch/odml_torch/passes/__init__.py sha256=AVwIwUTMx7rXacKjGy4kwrtMd3XB2v_ncdc40KOjUqQ 1245
ai_edge_torch/quantize/__init__.py sha256=aB5dXot04bqyUhpsDFvxt9CIi15QAC4euvqOndJ0XLU 714
ai_edge_torch/quantize/pt2e_quantizer.py sha256=CKIEhs9jCcna64qj1jFH9zEbMbRdyeGV_TmSqEBPjes 15741
ai_edge_torch/quantize/pt2e_quantizer_utils.py sha256=eARD1LxLi5m7Z0n_psAkeX_AtUp4fNkE--oECBfivv4 36208
ai_edge_torch/quantize/quant_config.py sha256=U0KisSW-uZkoMJcy-ZP9W57p3tsa594fr9PphCRdO8o 3172
ai_edge_torch/testing/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/testing/model_coverage/__init__.py sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE 765
ai_edge_torch/testing/model_coverage/model_coverage.py sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew 4718
ai_edge_torch_nightly-0.3.0.dev20241120.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
ai_edge_torch_nightly-0.3.0.dev20241120.dist-info/METADATA sha256=1Nv_QeerPRw888sOTf4jHx5Ihu-PJD9rL8GOpRHSTa4 1897
ai_edge_torch_nightly-0.3.0.dev20241120.dist-info/WHEEL sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo 92
ai_edge_torch_nightly-0.3.0.dev20241120.dist-info/top_level.txt sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE 14
ai_edge_torch_nightly-0.3.0.dev20241120.dist-info/RECORD — —

top_level.txt

ai_edge_torch