Searched refs:cudnn_gru (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 279 last_output, outputs, new_h, runtime = cudnn_gru(
301 api_name, _GPU_DEVICE_NAME, cudnn_gru)
392 def cudnn_gru(inputs, init_h, kernel, recurrent_kernel, bias, time_major): function
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 1157 def cudnn_gru(inputs, function
|