We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5d3ed88 commit 6ea4092Copy full SHA for 6ea4092
backends/apple/coreml/test/test_coreml_partitioner.py
@@ -71,23 +71,15 @@ def test_vit_skip_conv(self):
71
)
72
73
74
- conv_block = ["aten.convolution.default", "executorch_call_delegate"]
75
- safe_softmax_block = [
76
- "getitem",
77
78
79
80
- "aten.any.dim",
81
- "executorch_call_delegate",
82
- ]
83
- final_block = ["getitem"]
84
- total = conv_block + 12 * safe_softmax_block + final_block
85
-
86
assert [
87
node.target.__name__
88
for node in delegated_program_manager.exported_program().graph.nodes
89
if node.op == "call_function"
90
- ] == total
+ ] == [
+ "aten.convolution.default",
+ "executorch_call_delegate",
+ "getitem",
+ ]
91
92
def test_buffer(self):
93
embedding_dim = 3
0 commit comments