Solved by ptrblck in post #4 
              
              
                I guess you might be using an older torchvision release, since the operation is available via torchvision.ops.DeformConv2d and this PR added autocast support ~1 year ago. 
This code snippet works fine for me: 
from torchvision import ops

deform_layer = ops.DeformConv2d(in_channels=3, out_channels=6…