Home
last modified time | relevance | path

Searched defs:output_ptr_batch (Results 1 – 6 of 6) 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.cc428 float* output_ptr_batch = in EvalFloat() local
446 float* output_ptr_batch = in EvalFloat() local
474 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalFloat() local
500 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalFloat() local
601 float* output_ptr_batch = in EvalHybrid() local
625 float* output_ptr_batch = in EvalHybrid() local
660 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalHybrid() local
691 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.cc215 float* output_ptr_batch = GetTensorData<float>(output); in EvalFloat() local
248 float* output_ptr_batch = GetTensorData<float>(output); in EvalHybrid() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf.cc95 float* output_ptr_batch = output_ptr + b * num_units; in ApplyTimeWeightsBiasAndActivation() local
108 float* output_ptr_batch = output_ptr + b * num_units; in ApplyTimeWeightsBiasAndActivation() local
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc111 float* output_ptr_batch = output_ptr + b * num_units; in ApplyTimeWeightsBiasAndActivation() local
124 float* output_ptr_batch = output_ptr + b * num_units; in ApplyTimeWeightsBiasAndActivation() local