tvm icon indicating copy to clipboard operation
tvm copied to clipboard

Published 20 hours ago • verified publisher icon apache

[Relax][PyTorch] Add support for torchvision.ops.stochastic_depth

Open mshr-h opened this issue 1 year ago • 1 comments

As per title. torchvision.ops.stochastic_depth isn't a pytorch core ops, but some pytorch image classification models such as convnext, inception_v3, and maxvit_t are using it. So I think it's worth supporting it.

mshr-h avatar Aug 23 '24 11:08 mshr-h

@tvm-bot rerun

mshr-h avatar Aug 25 '24 04:08 mshr-h