-
- Downloads
CPP implementation of L2Norm and LRN ops (#1157)
Showing
- nnvm/include/nnvm/top/nn.h 35 additions, 0 deletionsnnvm/include/nnvm/top/nn.h
- nnvm/python/nnvm/top/nn.py 33 additions, 0 deletionsnnvm/python/nnvm/top/nn.py
- nnvm/src/top/nn/nn.cc 47 additions, 0 deletionsnnvm/src/top/nn/nn.cc
- nnvm/tests/python/compiler/test_top_level1.py 61 additions, 1 deletionnnvm/tests/python/compiler/test_top_level1.py
- topi/include/topi/cuda/normalization.h 106 additions, 0 deletionstopi/include/topi/cuda/normalization.h
- topi/include/topi/nn/l2_normalize.h 46 additions, 0 deletionstopi/include/topi/nn/l2_normalize.h
- topi/include/topi/nn/local_response_norm.h 76 additions, 0 deletionstopi/include/topi/nn/local_response_norm.h
- topi/include/topi/rocm/normalization.h 41 additions, 0 deletionstopi/include/topi/rocm/normalization.h
- topi/python/topi/cuda/__init__.py 1 addition, 1 deletiontopi/python/topi/cuda/__init__.py
- topi/python/topi/cuda/nn.py 11 additions, 57 deletionstopi/python/topi/cuda/nn.py
- topi/python/topi/generic/nn.py 10 additions, 7 deletionstopi/python/topi/generic/nn.py
- topi/python/topi/nn/__init__.py 1 addition, 1 deletiontopi/python/topi/nn/__init__.py
- topi/python/topi/nn/l2_normalize.py 29 additions, 0 deletionstopi/python/topi/nn/l2_normalize.py
- topi/python/topi/nn/local_response_norm.py 2 additions, 26 deletionstopi/python/topi/nn/local_response_norm.py
- topi/python/topi/rocm/nn.py 10 additions, 5 deletionstopi/python/topi/rocm/nn.py
- topi/python/topi/testing/__init__.py 2 additions, 0 deletionstopi/python/topi/testing/__init__.py
- topi/python/topi/testing/l2_normalize_python.py 27 additions, 0 deletionstopi/python/topi/testing/l2_normalize_python.py
- topi/python/topi/testing/lrn_python.py 53 additions, 0 deletionstopi/python/topi/testing/lrn_python.py
- topi/src/topi.cc 39 additions, 0 deletionstopi/src/topi.cc
- topi/tests/python/test_topi_l2norm.py 19 additions, 42 deletionstopi/tests/python/test_topi_l2norm.py
Loading
Please register or sign in to comment