Home
last modified time | relevance | path

Searched defs:output_ptr_batch (Results 1 – 4 of 4) 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/
Dbidirectional_sequence_rnn.cc430 float* output_ptr_batch = in EvalFloat() local
448 float* output_ptr_batch = in EvalFloat() local
476 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalFloat() local
502 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalFloat() local
603 float* output_ptr_batch = in EvalHybrid() local
627 float* output_ptr_batch = in EvalHybrid() local
662 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalHybrid() local
693 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalHybrid() local
Dunidirectional_sequence_rnn.cc236 float* output_ptr_batch = in EvalFloat() local
255 float* output_ptr_batch = GetTensorData<float>(output) + in EvalFloat() local
317 float* output_ptr_batch = in EvalHybrid() local
340 float* output_ptr_batch = GetTensorData<float>(output) + in EvalHybrid() local
Dbasic_rnn.cc216 float* output_ptr_batch = GetTensorData<float>(output); in EvalFloat() local
249 float* output_ptr_batch = GetTensorData<float>(output); in EvalHybrid() local