wozeparrot 2fd2fb6380 int8/uint8 support (#837) ...

* feat: int8 support

* feat: uint8 support

* feat: int8 tests

* fix: fix uint8 on clang

* feat: test casting between int8/uint8/float16/float32

* clean: way cleaner dtype tests

* feat: preprocess_imagenet using the correct dtype

* feat: add test for overflow between uint8 and int8

2023-05-28 23:15:06 -07:00

..

mnist

move datasets into datasets

2021-10-30 19:55:50 -07:00

__init__.py

start on mlperf models

2023-05-10 16:30:49 -07:00

imagenet.py

Cleanup mlperf (#797)

2023-05-25 11:36:43 -07:00

kits19.py

Add MLPerf UNet3D model (#775)

2023-05-28 20:38:19 -07:00

librispeech.py

Cleanup mlperf (#797)

2023-05-25 11:36:43 -07:00

openimages.py

add retinanet with resnet backbone (#813)

2023-05-28 20:20:16 -07:00

preprocess_imagenet.py

int8/uint8 support (#837)

2023-05-28 23:15:06 -07:00

squad.py

Add mlperf bert model (#803)

2023-05-27 14:53:32 -07:00