vision
vision copied to clipboard
deform_conv2d on MPS.
🚀 The feature
I want the deform_conv2d to be able to run on MPS(Apple Silicon) natively not falling back to CPU.
Motivation, pitch
I want to use this on MAC:
return torchvision.ops.deform_conv2d(x, offset, self.weight, self.bias, self.stride, self.padding, self.dilation, mask)
Alternatives
No response
Additional context
No response
Duplicate of deform_conv2d for mps #7490