chenyuandGitHub a65c8de735 move .half() llama freq_cis to the end of sin and cos (#4587) ...

otherwise arange has inf if either dim or context length exceeds half.max

2024-05-14 15:00:18 -04:00

..

bert.py

faster gather for bert (#4526)

2024-05-10 22:28:48 -07:00

convnext.py

move to new cached fetch (#2493)

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

efficientnet.py

simple LoadOps.ASSIGN (#3745)

2024-03-14 20:44:34 -07:00

llama.py

move .half() llama freq_cis to the end of sin and cos (#4587)

2024-05-14 15:00:18 -04:00

mask_rcnn.py

fix example that calls Tensor.__bool__ (#3650)

2024-03-07 16:59:26 -05:00

resnet.py

fp16 resnet (without expand backwards sum in float, doesn't work) (#3816)

2024-03-28 01:25:37 -04:00

retinanet.py

move to new cached fetch (#2493)

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

rnnt.py

create engine folder and move code (#3948)

2024-03-26 20:38:03 -07:00

transformer.py

replace with tensor op (#3099)

2024-01-12 14:13:40 -05: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