tinygrad/test/extra
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
..
test_export_model.py Allow multi-input model export (#1995) 2023-10-07 04:13:34 -07:00
test_extra_helpers.py apply view.py patch (#1844) 2023-09-10 17:32:15 -07:00
test_lr_scheduler.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
test_utils.py fixes (#1893) 2023-09-22 07:20:27 +08:00