Searched refs:aux_input_ptr_batch (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 37 const float* aux_input_ptr_batch, in RnnBatchStep() argument
59 aux_input_weights_ptr, num_units, aux_input_size, aux_input_ptr_batch, in RnnBatchStep()
93 aux_input_ptr_batch + k * aux_input_size, in RnnBatchStep()
142 float input_weights_scale, const float* aux_input_ptr_batch, in RnnBatchStep() argument
180 if (aux_input_ptr_batch && in RnnBatchStep()
181 !tensor_utils::IsZeroVector(aux_input_ptr_batch, in RnnBatchStep()
187 aux_input_ptr_batch + offset, aux_input_size, in RnnBatchStep()
259 if (aux_input_ptr_batch && in RnnBatchStep()
260 !tensor_utils::IsZeroVector(aux_input_ptr_batch, in RnnBatchStep()
266 aux_input_ptr_batch + offset, aux_input_size, in RnnBatchStep()
|
| D | kernel_utils.h | 48 const float* aux_input_ptr_batch,
77 float input_weights_scale, const float* aux_input_ptr_batch,
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 330 const float* aux_input_ptr_batch = in EvalFloat() local
338 input_ptr_batch, fw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
348 const float* aux_input_ptr_batch = in EvalFloat() local
358 input_ptr_batch, bw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
373 const float* aux_input_ptr_batch = in EvalFloat() local
380 input_ptr_batch, fw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
396 const float* aux_input_ptr_batch = in EvalFloat() local
403 input_ptr_batch, bw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
492 const float* aux_input_ptr_batch = in EvalHybrid() local
501 aux_input_ptr_batch, aux_fw_input_weights_ptr, in EvalHybrid()
[all …]
|
| D | lstm_eval.cc | 97 const float* input_to_output_weights_ptr, const float* aux_input_ptr_batch, in LstmStepWithAuxInput() argument
168 if (aux_input_ptr_batch != nullptr) { in LstmStepWithAuxInput()
172 aux_input_ptr_batch, n_batch, input_gate_scratch, in LstmStepWithAuxInput()
178 aux_input_ptr_batch, n_batch, forget_gate_scratch, /*result_stride=*/1); in LstmStepWithAuxInput()
180 aux_input_to_cell_weights_ptr, n_cell, n_aux_input, aux_input_ptr_batch, in LstmStepWithAuxInput()
184 aux_input_ptr_batch, n_batch, output_gate_scratch, /*result_stride=*/1); in LstmStepWithAuxInput()
437 float input_to_output_weights_scale, const float* aux_input_ptr_batch, in LstmStepWithAuxInput() argument
551 if (aux_input_ptr_batch != nullptr && in LstmStepWithAuxInput()
552 !tensor_utils::IsZeroVector(aux_input_ptr_batch, n_batch * n_input)) { in LstmStepWithAuxInput()
558 aux_input_ptr_batch + offset, n_input, in LstmStepWithAuxInput()
|