| 547 | #endif |
| 548 | |
| 549 | port::StatusOr<PersistentRnnPlan> CreatePersistentRnnPlan( |
| 550 | cudnnRNNDescriptor_t rnn_desc, int batch_size, cudnnDataType_t data_type) { |
| 551 | cudnnPersistentRNNPlan_t result; |
| 552 | RETURN_IF_CUDNN_ERROR( |
| 553 | cudnnCreatePersistentRNNPlan(rnn_desc, batch_size, data_type, &result)); |
| 554 | return port::StatusOr<PersistentRnnPlan>(PersistentRnnPlan(result)); |
| 555 | } |
| 556 | |
| 557 | // Turns a BatchDescriptor structure into a cudnn tensor handle within a |
| 558 | // scope. |