torchscript_consistency_cpu_test.py 185 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=['cpu'])