search

zakki / Ray

Computer Go Program. Download:https://github.com/zakki/Ray/releases
BSD 2-Clause "Simplified" License
149 stars 58 forks source link

Research on NNVM #137

Open zakki opened 6 years ago

zakki commented 6 years ago

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

trainewbie commented 6 years ago

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.)