Ray icon indicating copy to clipboard operation
Ray copied to clipboard

Research on NNVM

Open zakki opened this issue 8 years ago • 1 comments

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

zakki avatar Dec 04 '17 14:12 zakki

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 -

trainewbie avatar Dec 05 '17 12:12 trainewbie