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.cc29 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()
45 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()
126 float* hidden_state_ptr_batch, float* output_ptr_batch, in RnnBatchStep()
153 float* hidden_state_ptr_batch, float* output_ptr_batch, in RnnBatchStep()
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_rnn.cc230 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalFloat() local
248 float* hidden_state_ptr_batch = in EvalFloat() local
311 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalHybrid() local
333 float* hidden_state_ptr_batch = in EvalHybrid() local
Dbasic_rnn.cc213 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalFloat() local
246 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalHybrid() local