test_topi_pooling.py
15.9 KB
-
[Relay/Topi][Op] 1D Pooling (#4663) · 8a98a2e7* Added 1D pooling to Topi * Added 1D pooling relay op and tests. * Added onnx parsing and tests for maxpool1d and averagepool1d * formatting * moved partial import. * Fixed typo.
Josh Fromm committed