MIOpen
MIOpen copied to clipboard
Implement FractionalMaxPool
-
Added FractionalMaxpool2d and FractionalMaxpool3d.
-
Added driver test and gtest for FractionalMaxpool.
-
New API is guarded by MIOPEN_BETA_API macro.
-
Average over all cases:
-
FractionalMaxpool 2d
| Type | Forward | Backward |
|---|---|---|
| float16 | 3.84 | 5.46 |
| float32 | 3.61 | 5.36 |
| bfloat16 | 3.81 | 5.80 |
- FractionalMaxpool 3d
| Type | Forward | Backward |
|---|---|---|
| float16 | 4.69 | 9.83 |
| float32 | 3.56 | 9.99 |
| bfloat16 | 3.97 | 10.23 |