chenyu c0f76ed4ea transformer kvcache and mask have same dtype as input (#2771) ...

* transformer kvcache and mask have same dtype as input

* don't use `=0` in cstyle ternary where

* (bool)

* where float16 test

2023-12-14 22:41:51 -05:00

..

bert.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

convnext.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

efficientnet.py

add name support to fetch (#2407)

2023-11-23 14:16:17 -08:00

llama.py

transformer kvcache and mask have same dtype as input (#2771)

2023-12-14 22:41:51 -05:00

mask_rcnn.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

resnet.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

retinanet.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

rnnt.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

transformer.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

unet3d.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

vit.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00