TPU support
talhaanwarch opened this issue · comments
Are these functions especially version v2 and v3 are supported on GPU only? or they are supported on tpu also?
v1 and v2 uses pytorch native operators, which should work with tpu. As for v3 operators, they are specially written with cuda and can only work with gpu.