Home
last modified time | relevance | path

Searched refs:output_ptr_batch (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc27 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep() argument
33 hidden_state_ptr_batch, output_ptr_batch); in RnnBatchStep()
43 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep() argument
49 output_ptr_batch); in RnnBatchStep()
54 output_ptr_batch, /*result_stride=*/1); in RnnBatchStep()
60 batch_size, output_ptr_batch, /*result_stride=*/1); in RnnBatchStep()
66 batch_size, output_ptr_batch, /*result_stride=*/1); in RnnBatchStep()
70 output_ptr_batch, num_units * batch_size, activation, output_ptr_batch); in RnnBatchStep()
71 tensor_utils::CopyVector(output_ptr_batch, num_units * batch_size, in RnnBatchStep()
77 output_ptr_batch + k * output_batch_leading_dim); in RnnBatchStep()
[all …]
Dkernel_utils.h44 float* hidden_state_ptr_batch, float* output_ptr_batch);
54 float* hidden_state_ptr_batch, float* output_ptr_batch);
73 float* hidden_state_ptr_batch, float* output_ptr_batch);
85 float* hidden_state_ptr_batch, float* output_ptr_batch);
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc120 float* output_ptr_batch) { in LstmStepWithAuxInput()
306 n_batch, output_ptr_batch); in LstmStepWithAuxInput()
308 tensor_utils::ZeroVector(output_ptr_batch, n_batch * n_output); in LstmStepWithAuxInput()
312 n_batch, output_ptr_batch, /*result_stride=*/1); in LstmStepWithAuxInput()
314 tensor_utils::ClipVector(output_ptr_batch, n_batch * n_output, in LstmStepWithAuxInput()
315 params->proj_clip, output_ptr_batch); in LstmStepWithAuxInput()
319 output_ptr_batch); in LstmStepWithAuxInput()
321 tensor_utils::CopyVector(output_ptr_batch, n_batch * n_output, in LstmStepWithAuxInput()
329 output_ptr_batch + k * output_batch_leading_dim); in LstmStepWithAuxInput()
334 output_ptr_batch + k * output_batch_leading_dim, n_output); in LstmStepWithAuxInput()
[all …]
Dbidirectional_sequence_rnn.cc334 float* output_ptr_batch = in EvalFloat() local
341 params->activation, fw_hidden_state_ptr_batch, output_ptr_batch); in EvalFloat()
352 float* output_ptr_batch = in EvalFloat() local
361 params->activation, bw_hidden_state_ptr_batch, output_ptr_batch); in EvalFloat()
377 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalFloat() local
384 output_ptr_batch); in EvalFloat()
400 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalFloat() local
407 output_ptr_batch); in EvalFloat()
496 float* output_ptr_batch = in EvalHybrid() local
507 fw_hidden_state_ptr_batch, output_ptr_batch); in EvalHybrid()
[all …]
Dunidirectional_sequence_rnn.cc175 float* output_ptr_batch = output->data.f + s * num_units * batch_size; in EvalFloat() local
180 hidden_state_ptr_batch, output_ptr_batch); in EvalFloat()
191 float* output_ptr_batch = in EvalFloat() local
197 params->activation, hidden_state_ptr_batch, output_ptr_batch); in EvalFloat()
255 float* output_ptr_batch = output->data.f + s * num_units * batch_size; in EvalHybrid() local
262 hidden_state_ptr_batch, output_ptr_batch); in EvalHybrid()
273 float* output_ptr_batch = in EvalHybrid() local
281 scaling_factors_ptr, hidden_state_ptr_batch, output_ptr_batch); in EvalHybrid()
Dbasic_rnn.cc161 float* output_ptr_batch = output->data.f; in EvalFloat() local
170 params->activation, hidden_state_ptr_batch, output_ptr_batch); in EvalFloat()
193 float* output_ptr_batch = output->data.f; in EvalHybrid() local
214 hidden_state_ptr_batch, output_ptr_batch); in EvalHybrid()
Dsvdf.cc74 float* output_ptr_batch = output->data.f + b * num_units; in ApplyTimeWeightsBiasAndActivation() local
76 tensor_utils::ReductionSumVector(scratch_ptr_batch, output_ptr_batch, in ApplyTimeWeightsBiasAndActivation()
82 float* output_ptr_batch = output->data.f + b * num_units; in ApplyTimeWeightsBiasAndActivation() local
83 tensor_utils::ApplyActivationToVector(output_ptr_batch, num_units, in ApplyTimeWeightsBiasAndActivation()
84 activation, output_ptr_batch); in ApplyTimeWeightsBiasAndActivation()