Support GPU/CUDA inference in torchscript model
·
6 commits
to torchscript_gpu_v2.5
since this release
- Support GPU/CUDA inference in torchscript model
- Support post layer norm
- Support custom init value for weight_c