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 7e99148 commit 1438a9fCopy full SHA for 1438a9f
backends/qualcomm/builders/op_conv.py
@@ -127,7 +127,7 @@ def define_node(
127
128
filter_node = self.get_node(node.args[1])
129
filter_tensor = get_parameter(filter_node, self.edge_program)
130
- # weight of pytorch OIHW(conv2d) / IODHW(conv3d) or IOHW(conv_transpose2d) / IODHW(conv_transpose3d),
+ # weight of pytorch OIHW(conv2d) / OIDHW(conv3d) or IOHW(conv_transpose2d) / IODHW(conv_transpose3d),
131
# yet QNN is HWIO or DHWIO
132
is_transpose_conv = cast(bool, node.args[6])
133
if is_conv2d:
0 commit comments