Searched refs:rnn_input_mode (Results 1 – 1 of 1) sorted by relevance
153 TFRNNInputMode rnn_input_mode, DataType dtype) in CudnnRnnParameters() argument163 rnn_input_mode_(rnn_input_mode), in CudnnRnnParameters()169 static_cast<int>(rnn_input_mode), dtype}); in CudnnRnnParameters()245 Status ParseTFRNNInputMode(const string& str, TFRNNInputMode* rnn_input_mode) { in ParseTFRNNInputMode() argument247 *rnn_input_mode = TFRNNInputMode::kRNNLinearInput; in ParseTFRNNInputMode()250 *rnn_input_mode = TFRNNInputMode::kRNNSkipInput; in ParseTFRNNInputMode()253 *rnn_input_mode = TFRNNInputMode::kAutoSelect; in ParseTFRNNInputMode()481 TFRNNInputMode rnn_input_mode; member492 static_cast<int>(rnn_mode), static_cast<int>(rnn_input_mode), in DebugString()913 ParseTFRNNInputMode(str, &model_types_.rnn_input_mode)); in CudnnRNNKernelCommon()[all …]