MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ToMIOpenRnnInputMode

Function ToMIOpenRnnInputMode

tensorflow/stream_executor/rocm/rocm_dnn.cc:1551–1560  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1549}
1550
1551miopenRNNInputMode_t ToMIOpenRnnInputMode(dnn::RnnInputMode input_mode) {
1552 switch (input_mode) {
1553 case dnn::RnnInputMode::kRnnLinearSkip:
1554 return miopenRNNlinear;
1555 case dnn::RnnInputMode::kRnnSkipInput:
1556 return miopenRNNskip;
1557 default:
1558 LOG(FATAL) << "Invalid RNN input mode: " << static_cast<int>(input_mode);
1559 }
1560}
1561
1562miopenRNNDirectionMode_t ToMIOpenRnnDirectionMode(
1563 dnn::RnnDirectionMode direction_mode) {

Callers 1

createRnnDescriptorMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected