Searched refs:cudnn_lstm (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 632 last_output, outputs, new_h, new_c, runtime = cudnn_lstm(
649 api_name, _GPU_DEVICE_NAME, cudnn_lstm)
785 def cudnn_lstm(inputs, input_h, input_c, kernel, recurrent_kernel, bias, function
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 1044 def cudnn_lstm(inputs, function
|