| 89 | } |
| 90 | |
| 91 | const std::pair<vector<Tensor>, vector<Tensor>> RNN::Backward(int flag, |
| 92 | const vector<Tensor>& grads) { |
| 93 | vector<Tensor> param_grad; |
| 94 | vector<Tensor> data_grad; |
| 95 | LOG(FATAL) << "CPU RNN is not implemented!"; |
| 96 | return std::make_pair(data_grad, param_grad); |
| 97 | } |
| 98 | |
| 99 | void RNN::ToDevice(std::shared_ptr<Device> device) { |
| 100 | Layer::ToDevice(device); |
nothing calls this directly
no outgoing calls
no test coverage detected