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 baa2691 commit fc97a9bCopy full SHA for fc97a9b
backends/arm/tosa_utils.py
@@ -168,7 +168,7 @@ def is_consumer_node_depthwise_conv2d(node: Node):
168
return False
169
170
171
-def tosa_shape(shape: tuple[int | torch.SymInt], dim_order):
+def tosa_shape(shape, dim_order):
172
reordered = tuple([shape[dim] for dim in dim_order])
173
# Dynamic shapes in executorch are represented with torch.SymInt objects in the shapes,
174
# in TOSA we do not have this concept and instead use -1.
0 commit comments