Keras Frontend (#273)
* vgg16 success * remove six.PY2, use sys.version_info; convert_activation() accepts activation type name(str, e.g. 'relu') as input; * add convert_merge * fix convert_batchnorm; improve tests * fix lint * add numpy-style pad operator * deal with asymmetry padding * resnet50 success * fix pool_convert; xception passes test * update tvm * fix bias error; all tests pass * use > >, not >>
Showing
- nnvm/docs/top.rst 2 additions, 1 deletionnnvm/docs/top.rst
- nnvm/include/nnvm/top/nn.h 15 additions, 0 deletionsnnvm/include/nnvm/top/nn.h
- nnvm/python/nnvm/frontend/__init__.py 1 addition, 0 deletionsnnvm/python/nnvm/frontend/__init__.py
- nnvm/python/nnvm/frontend/common.py 35 additions, 0 deletionsnnvm/python/nnvm/frontend/common.py
- nnvm/python/nnvm/frontend/coreml.py 1 addition, 34 deletionsnnvm/python/nnvm/frontend/coreml.py
- nnvm/python/nnvm/frontend/keras.py 465 additions, 0 deletionsnnvm/python/nnvm/frontend/keras.py
- nnvm/python/nnvm/top/attr_dict.py 16 additions, 0 deletionsnnvm/python/nnvm/top/attr_dict.py
- nnvm/python/nnvm/top/nn.py 18 additions, 1 deletionnnvm/python/nnvm/top/nn.py
- nnvm/src/top/nn/nn.cc 37 additions, 0 deletionsnnvm/src/top/nn/nn.cc
- nnvm/tests/python/compiler/test_top_level1.py 20 additions, 0 deletionsnnvm/tests/python/compiler/test_top_level1.py
- nnvm/tests/python/frontend/keras/test_forward.py 117 additions, 0 deletionsnnvm/tests/python/frontend/keras/test_forward.py
Loading
Please register or sign in to comment