未验证 提交 9f61bc36 编写于 作者: G Guanghua Yu 提交者: GitHub

fix flatten in quant (#37722)

上级 8ac0344a
...@@ -136,7 +136,7 @@ _op_real_in_out_name = { ...@@ -136,7 +136,7 @@ _op_real_in_out_name = {
"flatten": [["X"], ["Out"]], "flatten": [["X"], ["Out"]],
"flatten2": [["X"], ["Out"]], "flatten2": [["X"], ["Out"]],
"unsqueeze2": [["X"], ["Out"]], "unsqueeze2": [["X"], ["Out"]],
"flatten_contiguous_range": [['X'], ["Out", "XShape"]], "flatten_contiguous_range": [['X'], ["Out"]],
} }
_conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose'] _conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册