Page MenuHomePhabricator

[mlir][tosa] Added shape propagation for TOSA pool operations.
ClosedPublic

Authored by rsuderman on Jul 8 2021, 3:18 PM.

Details

Summary

Pool operations perform the same shape propagation. Included the shape
propagation and tests for these avg_pool2d and max_pool2d.

Diff Detail

Event Timeline

rsuderman created this revision.Jul 8 2021, 3:18 PM
rsuderman requested review of this revision.Jul 8 2021, 3:18 PM

Rebased to head

sjarus accepted this revision.Jul 12 2021, 12:04 PM
This revision is now accepted and ready to land.Jul 12 2021, 12:04 PM
rsuderman updated this revision to Diff 358069.Jul 12 2021, 2:29 PM

Fixed clang-tidy error.