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


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