Solved by IvanKobzarev in post #2 
              
              
                The listing of supported ops you can find here: 

If operation is not supported - serialization will fail with exception pytorch/serializer.py at master · pytorch/pytorch · GitHub