Home
last modified time | relevance | path

Searched defs:hidden_state_ptr_batch (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc27 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()
43 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()
127 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()
150 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_rnn.cc169 float* hidden_state_ptr_batch = hidden_state->data.f; in EvalFloat() local
186 float* hidden_state_ptr_batch = hidden_state->data.f + b * num_units; in EvalFloat() local
249 float* hidden_state_ptr_batch = hidden_state->data.f; in EvalHybrid() local
268 float* hidden_state_ptr_batch = hidden_state->data.f + b * num_units; in EvalHybrid() local
Dbasic_rnn.cc158 float* hidden_state_ptr_batch = hidden_state->data.f; in EvalFloat() local
190 float* hidden_state_ptr_batch = hidden_state->data.f; in EvalHybrid() local