Searched defs:cell_state (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 292 TfLiteTensor* cell_state = in Prepare() local
487 TfLiteTensor* cell_state = in Eval() local
|
| D | lstm.cc | 367 TfLiteTensor* cell_state = in Prepare() local
572 TfLiteTensor* cell_state = in Eval() local
|
| D | lstm_eval.cc | 870 TfLiteStatus EvalFloat( in EvalFloat()
1068 TfLiteStatus EvalHybrid( in EvalHybrid()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2066 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
2175 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2212 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2259 def loop_fn(time_, cell_output, cell_state, _): argument
2328 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
|