Home
last modified time | relevance | path

Searched refs:CuDNNGRU (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dcudnn_recurrent_test.py41 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
57 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
72 ('cudnngru', keras.layers.CuDNNGRU),
97 ('cudnngru', keras.layers.CuDNNGRU),
123 ('cudnngru', keras.layers.CuDNNGRU),
159 ('cudnngru', keras.layers.CuDNNGRU),
191 ('cudnngru', keras.layers.CuDNNGRU),
249 for layer_class in [keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM]:
292 cudnn_rnn_layer_class = keras.layers.CuDNNGRU
372 cudnn_rnn_layer_class = keras.layers.CuDNNGRU
[all …]
Dcudnn_recurrent.py162 class CuDNNGRU(_CuDNNRNN): class
219 super(CuDNNGRU, self).__init__(
244 super(CuDNNGRU, self).build(input_shape)
334 base_config = super(CuDNNGRU, self).get_config()
D__init__.py167 from tensorflow.python.keras.layers.cudnn_recurrent import CuDNNGRU
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.layers.-cu-d-n-n-g-r-u.pbtxt1 path: "tensorflow.keras.layers.CuDNNGRU"
3 is_instance: "<class \'tensorflow.python.keras.layers.cudnn_recurrent.CuDNNGRU\'>"
Dtensorflow.keras.layers.pbtxt120 name: "CuDNNGRU"
/external/tensorflow/tensorflow/contrib/eager/python/examples/nmt_with_attention/
Dnmt_with_attention.ipynb390 " # If you have a GPU, we recommend using CuDNNGRU(provides a 3x speedup than GRU)\n",
393 " return tf.keras.layers.CuDNNGRU(units, \n",
/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format_test.py206 ('cudnngru', keras.layers.CuDNNGRU, {
/external/tensorflow/
DRELEASE.md376 * Added [`tf.keras.layers.CuDNNGRU`](https://www.tensorflow.org/versions/r1.9/api_docs/python/tf/ke…