| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 24 void RnnBatchStep(const float* input_ptr_batch, const float* input_weights_ptr, in RnnBatchStep() argument
30 RnnBatchStep(input_ptr_batch, input_weights_ptr, in RnnBatchStep()
38 void RnnBatchStep(const float* input_ptr_batch, const float* input_weights_ptr, in RnnBatchStep() argument
55 input_weights_ptr, num_units, input_size, input_ptr_batch, batch_size, in RnnBatchStep()
86 input_ptr_batch + k * input_size, /*n_batch=*/1, in RnnBatchStep()
120 const float* input_ptr_batch, const int8_t* input_weights_ptr, in RnnBatchStep() argument
129 RnnBatchStep(input_ptr_batch, input_weights_ptr, input_weights_scale, in RnnBatchStep()
144 const float* input_ptr_batch, const int8_t* input_weights_ptr, in RnnBatchStep() argument
189 if (!tensor_utils::IsZeroVector(input_ptr_batch, batch_size * input_size)) { in RnnBatchStep()
193 input_ptr_batch, batch_size, input_size, quantized_input_ptr_batch, in RnnBatchStep()
[all …]
|
| D | kernel_utils.h | 39 void RnnBatchStep(const float* input_ptr_batch, const float* input_weights_ptr,
47 void RnnBatchStep(const float* input_ptr_batch, const float* input_weights_ptr,
67 const float* input_ptr_batch, const int8_t* input_weights_ptr,
78 const float* input_ptr_batch, const int8_t* input_weights_ptr,
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 422 const float* input_ptr_batch = in EvalFloat() local
432 input_ptr_batch, fw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
440 const float* input_ptr_batch = in EvalFloat() local
453 input_ptr_batch, bw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
466 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalFloat() local
477 input_ptr_batch, fw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
492 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalFloat() local
503 input_ptr_batch, bw_input_weights_ptr, aux_input_ptr_batch, in EvalFloat()
595 const float* input_ptr_batch = in EvalHybrid() local
605 input_ptr_batch, fw_input_weights_ptr, fw_input_weights_scale, in EvalHybrid()
[all …]
|
| D | unidirectional_sequence_rnn.cc | 234 const float* input_ptr_batch = in EvalFloat() local
240 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
252 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalFloat() local
259 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
315 const float* input_ptr_batch = in EvalHybrid() local
321 input_ptr_batch, input_weights_ptr, input_weights_scale, in EvalHybrid()
337 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalHybrid() local
343 input_ptr_batch, input_weights_ptr, input_weights_scale, in EvalHybrid()
|
| D | basic_rnn.cc | 214 const float* input_ptr_batch = GetTensorData<float>(input); in EvalFloat() local
222 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
247 const float* input_ptr_batch = GetTensorData<float>(input); in EvalHybrid() local
270 input_ptr_batch, input_weights_ptr, input_weights_scale, in EvalHybrid()
|