Skip to content
Snippets Groups Projects
Commit c5f928bd authored by auphelia's avatar auphelia
Browse files

[CustomOp] Add new node attribute "dw" to Im2Col

parent e43f9c61
No related branches found
No related tags found
No related merge requests found
......@@ -79,6 +79,7 @@ class Im2Col(CustomOp):
"input_shape": ("s", True, ""),
"pad_amount": ("i", False, 0),
"pad_value": ("i", False, 0),
"dw": ("i", False, 0),
}
def make_shape_compatible_op(self, model):
......
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