Skip to content

GitLab

"test/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "766f8d1beda180ccb249a2623530d8be84fb9c22"
Switch branch/tag
Find file BlameHistoryPermalink
  • Charlie Lin's avatar
    Refactor Pooling and implement ONNX LpPool and GlobalLpPool (#1152) · 764273e4
    Charlie Lin authored 
    Refactored the reference implementation of pooling to something like what was done for roialign. Moved the reference implementation of pooling from targets/ref/lowering.cpp to pooling.hpp.
Removed cpu_pooling, instead using reference pooling in pooling.hpp
Added reference implementation of Lp Norm pooling and the global version
Added tests for the Lp Norm Pooling
    764273e4
pooling.cpp 1.4 KB