Page MenuHomePhabricator

[mlir][tosa] Fix tosa.avg_pool2d lowering to normalize correctly
ClosedPublic

Authored by rsuderman on May 14 2021, 4:39 PM.

Details

Summary

Initial version of pooling assumed normalization was accross all elements
equally. TOSA actually requires the noramalization is perform by how
many elements were summed (edges are not artifically dimmer). Updated
the lowering to reflect this change with corresponding tests.

Diff Detail

Event Timeline

rsuderman created this revision.May 14 2021, 4:39 PM
rsuderman requested review of this revision.May 14 2021, 4:39 PM
NatashaKnk accepted this revision.May 14 2021, 4:49 PM
This revision is now accepted and ready to land.May 14 2021, 4:49 PM