How to convert model with custom C++ CUDA layer to TorchScript? I have a model with Mixed C++/CUDA layer created like here: Custom C++ and CUDA Extensions — PyTorch Tutorials 1.9.0+cu102 documentation 2 .
I want to convert the whole model to TorchScript to use it in production. Is it possible to convert it? If yes, what is the best and proper way to do it?