MIOpen icon indicating copy to clipboard operation
MIOpen copied to clipboard
verified publisher icon ROCmSoftwarePlatform

Implement Avgpool

Open hieule88 opened this issue 10 months ago • 0 comments

  • Added AvgPool 2D 3D forward and backward with new solvers and bfp16 supported.

  • Added driver test and gtest for AvgPool.

  • New API is guarded by MIOPEN_BETA_API macro.

  • Average over all cases:

  • AvgPool 2D

Type Forward Backward
float16 1.70 1.88
float32 1.58 2.46
bfloat16 1.77 3.34
  • AvgPool 3D
Type Forward Backward
float16 1.53 1.31
float32 1.35 Not Improved
bfloat16 1.57 1.72

hieule88 avatar Mar 13 '25 03:03 hieule88