_build_rnn_graph
Yingting
1 def _build_rnn_graph(self, inputs, config, is_training):
2 if config.rnn_mode == CUDNN:
3 return self._build_rnn_graph_cudnn(inputs, config, is_training)
4 else:
5 return self._build_rnn_graph_lstm(inputs, config, is_training)