| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | gru_ops.h | 30 struct GRUCell { struct
31 GRUCell(const int batch_size, const int input_size, const int cell_size) in GRUCell() argument
69 struct GRUBlockCellFprop : public GRUCell { argument
72 : GRUCell(batch_size, input_size, cell_size) {} in GRUBlockCellFprop()
125 struct GRUBlockCellBprop : public GRUCell {
128 : GRUCell(batch_size, input_size, cell_size) {} in GRUBlockCellBprop()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 44 GRUCell = rnn_cell_impl.GRUCell variable
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | unfused_gru.py | 43 cell_fw = tf.compat.v1.nn.rnn_cell.GRUCell(parameters["units"])
44 cell_bw = tf.compat.v1.nn.rnn_cell.GRUCell(parameters["units"])
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | rnn_cell_wrapper_v2_test.py | 47 base_cell = rnn_cell_impl.GRUCell(
71 base_cell = rnn_cell_impl.GRUCell(
94 cell = rnn_cell_impl.GRUCell(3)
201 cell = layers.GRUCell(10)
|
| D | __init__.py | 225 from tensorflow.python.keras.layers.recurrent_v2 import GRUCell
229 from tensorflow.python.keras.layers.recurrent import GRUCell as GRUCellV1
233 GRUCellV2 = GRUCell
238 from tensorflow.python.keras.layers.recurrent import GRUCell
242 from tensorflow.python.keras.layers.recurrent_v2 import GRUCell as GRUCellV2
246 GRUCellV1 = GRUCell
|
| D | gru_test.py | 221 cell = keras.layers.GRUCell(20)
258 cell = keras.layers.GRUCell(5)
|
| D | recurrent_test.py | 426 cells = [keras.layers.recurrent.GRUCell(8),
444 cells = [keras.layers.recurrent.GRUCell(32, reset_after=True)]
487 cells = [keras.layers.recurrent.GRUCell(8),
564 cells = [keras.layers.recurrent.GRUCell(8),
704 keras.layers.GRUCell,
777 cell=[keras.layers.SimpleRNNCell, keras.layers.GRUCell,
1428 [keras.layers.SimpleRNNCell, keras.layers.GRUCell, keras.layers.LSTMCell])
1552 [keras.layers.SimpleRNNCell, keras.layers.GRUCell, keras.layers.LSTMCell])
1733 [rnn_v1.SimpleRNNCell, rnn_v1.GRUCell, rnn_v1.LSTMCell, rnn_v2.GRUCell,
|
| D | recurrent_v2.py | 109 class GRUCell(recurrent.GRUCell): class
199 super(GRUCell, self).__init__(
|
| D | recurrent.py | 1690 class GRUCell(DropoutRNNCellMixin, Layer): class
1761 super(GRUCell, self).__init__(**kwargs)
1957 base_config = super(GRUCell, self).get_config()
2088 cell = GRUCell(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.rnn_cell.pbtxt | 20 name: "GRUCell"
|
| D | tensorflow.keras.layers.-g-r-u-cell.pbtxt | 1 path: "tensorflow.keras.layers.GRUCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.GRUCell\'>"
|
| D | tensorflow.nn.rnn_cell.-g-r-u-cell.pbtxt | 1 path: "tensorflow.nn.rnn_cell.GRUCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.legacy_rnn.rnn_cell_impl.GRUCell\'>"
|
| D | tensorflow.keras.layers.pbtxt | 180 name: "GRUCell"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-g-r-u-cell.pbtxt | 1 path: "tensorflow.keras.layers.GRUCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent_v2.GRUCell\'>"
4 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.GRUCell\'>"
|
| D | tensorflow.keras.layers.pbtxt | 172 name: "GRUCell"
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | legacy_rnn_test.py | 99 cell = tf.keras.layers.GRUCell(output_shape)
316 tf.nn.rnn_cell.GRUCell(32, dtype=tf.float32)
337 "GRUCell": tf.nn.rnn_cell.GRUCell,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 948 cell = rnn_cell.GRUCell(30)
1991 cell = rnn_cell.GRUCell(num_units=num_units)
2039 cell = rnn_cell.GRUCell(num_units=num_units)
2553 g, _ = rnn_cell_impl.GRUCell(2)(x, m)
2566 g, _ = rnn_cell_impl.GRUCell(2)(x, m)
2832 base_cell = rnn_cell_impl.GRUCell(
2855 base_cell = rnn_cell_impl.GRUCell(
2878 cell = rnn_cell_impl.GRUCell(3)
2910 cell = rnn_cell_impl.DeviceWrapper(rnn_cell_impl.GRUCell(3), gpu_dev)
2933 [rnn_cell_impl.GRUCell(2) for _ in range(2)],
[all …]
|
| D | rnn_test.py | 154 rnn_cell_impl.GRUCell,
329 self._assert_cell_builds(rnn_cell_impl.GRUCell, f32, 5, 7, 3)
330 self._assert_cell_builds(rnn_cell_impl.GRUCell, f64, 5, 7, 3)
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 495 class GRUCell(LayerRNNCell): class
539 super(GRUCell, self).__init__(
626 base_config = super(GRUCell, self).get_config()
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | integration_test.py | 226 model.add(keras.layers.RNN(rnn_cell.GRUCell(y_train.shape[-1],
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_NumPy_Text_Generation.ipynb | 485 "class GRUCell:\n",
531 " self._cell = GRUCell(n_units, forget_bias)\n",
|
| /external/tensorflow/ |
| D | RELEASE.md | 3840 `RNNCell`, `LSTMCell`, `GRUCell`, and a number of other cells. These
|