github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity
Files
8d6cecb25c7f1e228494bc1fb2c786512b82334f
tinygrad/extra/export_model.py
Ahmed Harmouche 2114dc13d1 Allow multi-input model export (#1995) 
* Allow multi-input model export

* Add model export unit test

* Fix efficientnet compilation

* Only run model export test on JIT supported devices

* Skip export model test if not EXPORT_SUPPORTED_DEVICE
2023-10-07 04:13:34 -07:00

8.7 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink