torchscript_consistency_cuda_test.py 186 Bytes
Newer Older
1
2
from .common_utils import define_test_suites
from .torchscript_consistency_impl import Functional, Transforms
3
4
5


define_test_suites(globals(), [Functional, Transforms], devices=['cuda'])