-
- Downloads
[TOPI] Add ops compute (#323)
* [TOPI] Add ops compute Remove 'compute' and add assert for safety Add document fix lint fix softmax * fix batch norm
Showing
- topi/python/topi/math.py 36 additions, 0 deletionstopi/python/topi/math.py
- topi/python/topi/nn/__init__.py 8 additions, 3 deletionstopi/python/topi/nn/__init__.py
- topi/python/topi/nn/batch_norm.py 55 additions, 0 deletionstopi/python/topi/nn/batch_norm.py
- topi/python/topi/nn/convolution.py 1 addition, 1 deletiontopi/python/topi/nn/convolution.py
- topi/python/topi/nn/elemwise.py 0 additions, 0 deletionstopi/python/topi/nn/elemwise.py
- topi/python/topi/nn/flatten.py 32 additions, 0 deletionstopi/python/topi/nn/flatten.py
- topi/python/topi/nn/fully_connected.py 62 additions, 0 deletionstopi/python/topi/nn/fully_connected.py
- topi/python/topi/nn/pooling.py 77 additions, 0 deletionstopi/python/topi/nn/pooling.py
- topi/python/topi/nn/softmax.py 27 additions, 0 deletionstopi/python/topi/nn/softmax.py
Loading
Please register or sign in to comment