-
- Downloads
Generalize pooling to support arbitrary layout (#1103)
* generalize pool2d to arbitrary layout * explain more the layout support for pool * allow missing factor size for pooling * explain what factor size is used for * fix typo * name idx -> axis
Showing
- topi/include/topi/cuda/pooling.h 3 additions, 1 deletiontopi/include/topi/cuda/pooling.h
- topi/include/topi/nn/pooling.h 145 additions, 153 deletionstopi/include/topi/nn/pooling.h
- topi/python/topi/cuda/pooling.py 2 additions, 1 deletiontopi/python/topi/cuda/pooling.py
- topi/python/topi/nn/pooling.py 46 additions, 192 deletionstopi/python/topi/nn/pooling.py
- topi/python/topi/opengl/pooling.py 2 additions, 1 deletiontopi/python/topi/opengl/pooling.py
- topi/python/topi/x86/injective.py 4 additions, 0 deletionstopi/python/topi/x86/injective.py
Loading
Please register or sign in to comment