"lm_eval/tasks/tinyBenchmarks/utils_winogrande.py" did not exist on "f7dde0c3ad239b3d4538aa92000686c2ec4dfaf2"
  • 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
lowering.cpp 16.7 KB