[torchaudio] Use more specific includes than torch/script.h (#3198)

Summary:
Pull Request resolved: https://github.com/pytorch/audio/pull/3198

Fixes build after following diff to use F14 maps in pickler internally.

Reviewed By: mthrok

Differential Revision: D44098387

fbshipit-source-id: 9777517369d9a3f2599b273c04bf4a014f411f12

torchaudio/csrc/rnnt/gpu/compute.cu

 #include <c10/cuda/CUDAStream.h>
-#include <torch/script.h>
+#include <torch/types.h>
 #include <torchaudio/csrc/rnnt/gpu/gpu_transducer.h>
 
 namespace torchaudio {

torchaudio/csrc/rnnt/gpu/compute_alphas.cu

 #include <c10/cuda/CUDAStream.h>
-#include <torch/script.h>
+#include <torch/types.h>
 #include <torchaudio/csrc/rnnt/gpu/gpu_transducer.h>
 
 namespace torchaudio {

torchaudio/csrc/rnnt/gpu/compute_betas.cu

 #include <c10/cuda/CUDAStream.h>
-#include <torch/script.h>
+#include <torch/types.h>
 #include <torchaudio/csrc/rnnt/gpu/gpu_transducer.h>
 
 namespace torchaudio {