| 1560 | } |
| 1561 | |
| 1562 | miopenRNNDirectionMode_t ToMIOpenRnnDirectionMode( |
| 1563 | dnn::RnnDirectionMode direction_mode) { |
| 1564 | switch (direction_mode) { |
| 1565 | case dnn::RnnDirectionMode::kRnnUnidirectional: |
| 1566 | return miopenRNNunidirection; |
| 1567 | case dnn::RnnDirectionMode::kRnnBidirectional: |
| 1568 | return miopenRNNbidirection; |
| 1569 | default: |
| 1570 | LOG(FATAL) << "Invalid RNN direction mode: " |
| 1571 | << static_cast<int>(direction_mode); |
| 1572 | } |
| 1573 | } |
| 1574 | |
| 1575 | miopenRNNMode_t ToMIOpenRnnMode(dnn::RnnMode rnn_mode) { |
| 1576 | switch (rnn_mode) { |