Skip to content
Snippets Groups Projects
Commit 91c97a52 authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Fixed operator names when exporting to hpvm config

parent 0e719db5
No related branches found
No related tags found
No related merge requests found
......@@ -253,7 +253,13 @@ class HPVMConfigBuilder:
["maxpool"],
]
op_to_op = {"convolution": "conv", "maxpool": "pool_max", "linear": "mul"}
op_to_op = {
"convolution": "conv",
"maxpool": "pool_max",
"linear": "mul",
"avgpool": "pool_mean",
"depthwise_convolution": "group_conv"
}
knob_name_to_range = {
"fp32": range(11, 12),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment