-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support some ops for full quantization #40083
Support some ops for full quantization #40083
Conversation
Thanks for your contribution! |
"hard_sigmoid", "conv2d_transpose", "gru", "bilinear_interp", | ||
"nearest_interp", "trilinear_interp", "flatten", "flatten2", "transpose", | ||
"pad2d", "reshape", "layer_norm", "split", "flatten_contiguous_range", | ||
"squeeze2", "nearest_interp_v2", "fill_constant_batch_size_like" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
list后边加个逗号,修改文件后的diff会更直观。
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
@@ -1804,7 +1780,8 @@ class AddQuantDequantPass(object): | |||
"squeeze", "elementwise_sub", "mul", "matmul", "relu", "relu6", | |||
"leaky_relu", "tanh", "swish", "scale", "transpose", "transpose2", | |||
"sigmoid", "pad2d", "flatten", "flatten2", "batch_norm", "layer_norm", | |||
"matmul_v2" | |||
"matmul_v2", "split", "flatten_contiguous_range", "squeeze2", | |||
"nearest_interp_v2", "fill_constant_batch_size_like" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
加逗号
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
@@ -137,6 +109,10 @@ | |||
"flatten2": [["X"], ["Out"]], | |||
"unsqueeze2": [["X"], ["Out"]], | |||
"flatten_contiguous_range": [['X'], ["Out"]], | |||
"split": [['X'], ["Out"]], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
split需要特殊处理下么?类似于concat。
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
out_threshold只输出一个即可,其余暂不需要特殊处理。
Sorry to inform you that 423912c's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* add some op for full_quantization
PR types
Function optimization
PR changes
APIs
Describe
Support some op for full quantization