Skip to content

Commit

Permalink
ONNX codegen
Browse files Browse the repository at this point in the history
  • Loading branch information
maheshambule committed May 20, 2020
1 parent 44d417e commit 406807f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/contrib/codegen_onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ class MaxPool(OpConverter):
@classmethod
def convert_attributes(cls, attrs):
return {
'pads': attrs.get_int_tuple("padding") + attrs.get_int_tuple("padding"),
'pads': attrs.get_int_tuple("padding"),
'strides': attrs.get_int_tuple("strides"),
'kernel_shape': attrs.get_int_tuple("pool_size"),
}
Expand Down

0 comments on commit 406807f

Please sign in to comment.