Skip to content
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

Implement AdaptiveMaxpool #3608

Open
wants to merge 11 commits into
base: develop
Choose a base branch
from
Open

Implement AdaptiveMaxpool #3608

wants to merge 11 commits into from

Conversation

hieule88
Copy link
Collaborator

  • Added AdaptiveMaxPool 1D 2D 3D forward and backward.

  • Added driver test and gtest for AdaptiveMaxPool .

  • New API is guarded by MIOPEN_BETA_API macro.

  • Average over all cases:

  • AdaptiveMaxPool 1D

Type Forward Backward
float16 8.97 4.62
float32 7.02 4.54
bfloat16 9.18 4.54
  • AdaptiveMaxPool 2D
Type Forward Backward
float16 5.84 2.41
float32 4.76 3.49
bfloat16 4.61 2.28
  • AdaptiveMaxPool 3D
Type Forward Backward
float16 3.90 2.16
float32 3.72 2.80
bfloat16 4.22 2.32

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant