[FRONTEND][TENSORFLOW] GPU support for tensorflow models. (#1718)
Showing
- nnvm/python/nnvm/frontend/tensorflow.py 64 additions, 26 deletionsnnvm/python/nnvm/frontend/tensorflow.py
- nnvm/tests/python/frontend/tensorflow/test_forward.py 20 additions, 8 deletionsnnvm/tests/python/frontend/tensorflow/test_forward.py
- tutorials/nnvm/from_tensorflow.py 13 additions, 5 deletionstutorials/nnvm/from_tensorflow.py
Loading
Please register or sign in to comment