Searched refs:TFRNNInputMode (Results 1 – 1 of 1) sorted by relevance
114 enum class TFRNNInputMode { enum153 TFRNNInputMode rnn_input_mode, DataType dtype) in CudnnRnnParameters()199 RnnMode, TFRNNInputMode, DataType>;216 const TFRNNInputMode rnn_input_mode_;245 Status ParseTFRNNInputMode(const string& str, TFRNNInputMode* rnn_input_mode) { in ParseTFRNNInputMode()247 *rnn_input_mode = TFRNNInputMode::kRNNLinearInput; in ParseTFRNNInputMode()250 *rnn_input_mode = TFRNNInputMode::kRNNSkipInput; in ParseTFRNNInputMode()253 *rnn_input_mode = TFRNNInputMode::kAutoSelect; in ParseTFRNNInputMode()271 Status ToRNNInputMode(TFRNNInputMode tf_input_mode, int num_units, in ToRNNInputMode()274 case TFRNNInputMode::kRNNLinearInput: in ToRNNInputMode()[all …]