File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
backends/xnnpack/test/models Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change 8
8
9
9
import torch
10
10
import torchvision .models as models
11
- from executorch .backends .xnnpack .partition .xnnpack_partitioner import (
12
- XnnpackQuantizedPartitioner ,
13
- )
14
- from executorch .backends .xnnpack .test .tester import Export , Partition , Tester
11
+ from executorch .backends .xnnpack .test .tester import Export , Tester
15
12
from executorch .exir import CaptureConfig
16
13
from torchvision .models .mobilenetv2 import MobileNet_V2_Weights
17
14
@@ -59,7 +56,7 @@ def test_qs8_mv2(self):
59
56
.export (Export (CaptureConfig (enable_aot = True )))
60
57
.to_edge ()
61
58
.check (list (ops_after_quantization ))
62
- .partition (Partition ( partitioner = XnnpackQuantizedPartitioner ) )
59
+ .partition ()
63
60
.check (["torch.ops.executorch_call_delegate" ])
64
61
.check_not (list (ops_after_quantization ))
65
62
.to_executorch ()
You can’t perform that action at this time.
0 commit comments