4 Commits

Author	SHA1	Message	Date
Francois Chollet	b11400f2fb	Fix various issues	2023-05-13 20:17:18 -07:00
Francois Chollet	fa2ea78d3f	Fix typos	2023-05-02 22:44:46 -07:00
Chen Qian	a85647b3ec	Implement global pooling layer (#74) ... * Add max and poolig layer * fix tests * handle TF transpose * renaming * initial * add something * rename tests * more * add docstring * add mask for global average pooling 1d	2023-05-02 22:40:42 -07:00
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