* 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
* Add max and poolig layer * fix tests * handle TF transpose * renaming * rename tests * Fix comments * Move out the shape computation logic