Searched refs:rnn_input_mode (Results 1 – 1 of 1) sorted by relevance
156 TFRNNInputMode rnn_input_mode, DataType dtype) in CudnnRnnParameters() argument166 rnn_input_mode_(rnn_input_mode), in CudnnRnnParameters()172 static_cast<int>(rnn_input_mode), dtype}); in CudnnRnnParameters()248 Status ParseTFRNNInputMode(const string& str, TFRNNInputMode* rnn_input_mode) { in ParseTFRNNInputMode() argument250 *rnn_input_mode = TFRNNInputMode::kRNNLinearInput; in ParseTFRNNInputMode()253 *rnn_input_mode = TFRNNInputMode::kRNNSkipInput; in ParseTFRNNInputMode()256 *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()1029 ParseTFRNNInputMode(str, &model_types_.rnn_input_mode)); in CudnnRNNKernelCommon()[all …]