Searched refs:cell_fw (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | unfused_gru.py | 43 cell_fw = tf.compat.v1.nn.rnn_cell.GRUCell(parameters["units"])
46 cell_fw, cell_bw, inputs, dtype=tf.float32)
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 282 def bidirectional_dynamic_rnn(cell_fw, argument
367 rnn_cell_impl.assert_like_rnncell("cell_fw", cell_fw)
374 cell=cell_fw,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 347 def bidirectional_dynamic_rnn(cell_fw, argument
432 rnn_cell_impl.assert_like_rnncell("cell_fw", cell_fw)
439 cell=cell_fw,
1520 def static_bidirectional_rnn(cell_fw, argument
1569 rnn_cell_impl.assert_like_rnncell("cell_fw", cell_fw)
1580 cell_fw,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_variable_test.py | 528 cell_fw = rnn_cell_impl.LSTMCell(300)
531 cell_fw, cell_bw, inputs, dtype=dtypes.float32)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 53 …argspec: "args=[\'cell_fw\', \'cell_bw\', \'inputs\', \'sequence_length\', \'initial_state_fw\', \…
393 …argspec: "args=[\'cell_fw\', \'cell_bw\', \'inputs\', \'initial_state_fw\', \'initial_state_bw\', …
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 1348 cell_fw = rnn_cell.LSTMCell(
1358 cell_fw,
1472 cell_fw = rnn_cell.LSTMCell(
1485 cell_fw,
|