Commit Graph

1 Commits

Author SHA1 Message Date
Chen Qian
841b8d702d Add max and average pooling layer (#66)
* Add max and poolig layer

* fix tests

* handle TF transpose

* renaming

* rename tests

* Fix comments

* Move out the shape computation logic
2023-05-02 15:17:59 -07:00