82
edits
Changes
Jump to navigation
Jump to search
no edit summary
'''
if tf.test.is_gpu_available():
lstm = Cudnn_LSTM_with_biastf.contrib.cudnn_rnn.CudnnCompatibleLSTMCell(num_units)
else:
lstm = tf.nn.rnn_cell.LSTMCell(num_units,forget_bias=1.0)
