1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
import torch
from torchaudio_unittest import common_utils
from .torchscript_consistency_impl import Functional, Transforms
class TestFunctionalFloat32(Functional, common_utils.PytorchTestCase):
dtype = torch.float32
device = torch.device('cpu')
class TestFunctionalFloat64(Functional, common_utils.PytorchTestCase):
dtype = torch.float64
device = torch.device('cpu')
class TestTransformsFloat32(Transforms, common_utils.PytorchTestCase):
dtype = torch.float32
device = torch.device('cpu')
class TestTransformsFloat64(Transforms, common_utils.PytorchTestCase):
dtype = torch.float64
device = torch.device('cpu')
|
