Add a testcase for ts compile -> saving with torch_tensorrt.save

Add a testcase for ts compile -> saving with torch_tensorrt.save #5121