Was trying to find a GPU implementation of dnn::forward .... wanted to know if there is any work being done?? .... and would the workaround this would be to get all the weights and write a cuda version of forward pass myself??
1 | initial version |
Was trying to find a GPU implementation of dnn::forward .... wanted to know if there is any work being done?? .... and would the workaround this would be to get all the weights and write a cuda version of forward pass myself??