Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Research on NNVM #137

Open
zakki opened this issue Dec 4, 2017 · 1 comment
Open

Research on NNVM #137

zakki opened this issue Dec 4, 2017 · 1 comment

Comments

@zakki
Copy link
Owner

zakki commented Dec 4, 2017

For AMD GPU and OpenCL via ONNX.
https://github.com/dmlc/nnvm

@trainewbie
Copy link

Perhaps, I think it is probably possible.
You can use the method to extract the parameters into a separate file and load it.
The reason I said "Perhaps"....
In the case of RNN, the implementation method differs for each vendor,
it may not be possible to extract the parameters.
Note that cuDNN_LSTM and Tensorflow BasicLSTM are different implementation methods and should be studied again.
(Of course, native cuDNN_LSTM is supported as a separate API in tensor flow.)
- I write with some programmer's advice -

zakki pushed a commit that referenced this issue May 31, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants