Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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
H A 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