-
- Downloads
[TOPI] Isolate padding option, improve decl of depthwise/conv2d/pool (#332)
Showing
- python/tvm/expr.py 6 additions, 0 deletionspython/tvm/expr.py
- python/tvm/make.py 5 additions, 4 deletionspython/tvm/make.py
- topi/python/topi/__init__.py 1 addition, 0 deletionstopi/python/topi/__init__.py
- topi/python/topi/nn/convolution.py 45 additions, 94 deletionstopi/python/topi/nn/convolution.py
- topi/python/topi/nn/dilate.py 18 additions, 19 deletionstopi/python/topi/nn/dilate.py
- topi/python/topi/nn/pad.py 104 additions, 0 deletionstopi/python/topi/nn/pad.py
- topi/python/topi/nn/pooling.py 18 additions, 19 deletionstopi/python/topi/nn/pooling.py
- topi/python/topi/nn/softmax.py 5 additions, 6 deletionstopi/python/topi/nn/softmax.py
- topi/python/topi/util.py 16 additions, 0 deletionstopi/python/topi/util.py
- topi/recipe/conv/depthwise_conv2d_test.py 1 addition, 1 deletiontopi/recipe/conv/depthwise_conv2d_test.py
- topi/tests/python/test_topi_depthwise_conv2d.py 1 addition, 1 deletiontopi/tests/python/test_topi_depthwise_conv2d.py
- topi/tests/python/test_topi_dilate.py 1 addition, 3 deletionstopi/tests/python/test_topi_dilate.py
Loading
Please register or sign in to comment