| 1573 | } |
| 1574 | |
| 1575 | miopenRNNMode_t ToMIOpenRnnMode(dnn::RnnMode rnn_mode) { |
| 1576 | switch (rnn_mode) { |
| 1577 | case dnn::RnnMode::kRnnRelu: |
| 1578 | return miopenRNNRELU; |
| 1579 | case dnn::RnnMode::kRnnTanh: |
| 1580 | return miopenRNNTANH; |
| 1581 | case dnn::RnnMode::kRnnLstm: |
| 1582 | return miopenLSTM; |
| 1583 | case dnn::RnnMode::kRnnGru: |
| 1584 | return miopenGRU; |
| 1585 | default: |
| 1586 | LOG(FATAL) << "Invalid RNN Mode: " << static_cast<int>(rnn_mode); |
| 1587 | } |
| 1588 | } |
| 1589 | |
| 1590 | int MIOpenDataTypeToByteSize(miopenDataType_t data_type) { |
| 1591 | switch (data_type) { |