Francis Lata 2c3417dfce Merge branch 'master' into retinanet_mlperf

2025-02-23 21:23:28 +00:00

..

bert.py

bert use int32 for input ids (#9173)

2025-02-19 08:17:27 -05:00

clip.py

clip device fix (#6924)

2024-10-07 00:47:32 +08:00

convnext.py

move to new cached fetch (#2493)

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

efficientnet.py

remove the magic methods for moving between devices [pr] (#6881)

2024-10-04 20:27:52 +08:00

inception.py

Compute FID Score (#6802)

2024-10-01 19:47:58 -04:00

llama.py

workaround f16 cast ambiguity (#8935)

2025-02-11 09:38:56 +08:00

mask_rcnn.py

Merge branch 'master' into retinanet_mlperf

2024-12-23 08:30:46 -08:00

resnet.py

no need to return loaded keys for resnet

2025-02-19 20:35:03 +00:00

retinanet.py

update loss calculation for regresionhead and some cleanups

2025-02-23 21:22:33 +00:00

rnnt.py

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04:00

t5.py

Flux.1 (#6334)

2024-09-24 10:08:04 +08:00

transformer.py

_one_hot_along_dim input needs to be int (#9179)

2025-02-20 09:00:43 -05:00

unet3d.py

move to new cached fetch (#2493)

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

unet.py

These casts should only happen if these are supported (#7644)

2024-11-12 07:56:50 +08:00

vit.py

move to new cached fetch (#2493)

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