Searched refs:gpu_gru (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 531 last_output, outputs, new_h, runtime = gpu_gru(**gpu_gru_kwargs)
627 def gpu_gru(inputs, init_h, kernel, recurrent_kernel, bias, mask, time_major, function
768 return gpu_gru(
780 return gpu_gru(
|