Skip to content
This repository was archived by the owner on Oct 13, 2021. It is now read-only.

Add unit tests for upsample and pool #99

Merged
merged 15 commits into from
Jun 4, 2019
Merged

Conversation

jiafatom
Copy link
Collaborator

@jiafatom jiafatom commented Jun 3, 2019

No description provided.

@jiafatom jiafatom requested a review from vinitra-zz June 3, 2019 22:12
@@ -46,7 +38,7 @@ def convert_keras_upsample(scope, operator, container, n_dims):

# If channels_first is True, we don't need to permute the output of ONNX Upsample. Otherwise, similar to Crop's
# conversion, a Transpose would be added.

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

update channels_first comments to indicate change to no_permutation_required

attrs['op_version'] = op_version
# In ONNX opset 10, the ceil_mode attribute was added to local MaxPool and AveragePool
if container.target_opset >= 10:
attrs['ceil_mode'] = 0

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for adding the ceil_mode opset10 changes! I can close my PR #96.

@vinitra-zz
Copy link

vinitra-zz commented Jun 3, 2019

LGTM, after it passes the CI tests! Thanks for merging my changes from #96 into this PR. Will focus solely on thresholdedReLU in that PR.

vinitra-zz pushed a commit to vinitra-zz/keras-onnx that referenced this pull request Jun 3, 2019
@jiafatom jiafatom merged commit f4731c3 into onnx:master Jun 4, 2019
@jiafatom jiafatom deleted the test_upsample branch June 19, 2019 23:04
wenbingl pushed a commit that referenced this pull request Oct 22, 2019
fix batch norm mean/var dimention issue
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants