Home
last modified time | relevance | path

Searched defs:input_ptr_batch (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_rnn.cc424 const float* input_ptr_batch = in EvalFloat() local
442 const float* input_ptr_batch = in EvalFloat() local
468 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalFloat() local
494 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalFloat() local
597 const float* input_ptr_batch = in EvalHybrid() local
621 const float* input_ptr_batch = in EvalHybrid() local
654 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalHybrid() local
685 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalHybrid() local
H A Dunidirectional_sequence_rnn.cc234 const float* input_ptr_batch = in EvalFloat() local
252 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalFloat() local
315 const float* input_ptr_batch = in EvalHybrid() local
337 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalHybrid() local
H A Dbasic_rnn.cc215 const float* input_ptr_batch = GetTensorData<float>(input); in EvalFloat() local
248 const float* input_ptr_batch = GetTensorData<float>(input); in EvalHybrid() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dkernel_utils.cc24 void RnnBatchStep(const float* input_ptr_batch, const float* input_weights_ptr, in RnnBatchStep()
38 void RnnBatchStep(const float* input_ptr_batch, const float* input_weights_ptr, in RnnBatchStep()
120 const float* input_ptr_batch, const int8_t* input_weights_ptr, in RnnBatchStep()
144 const float* input_ptr_batch, const int8_t* input_weights_ptr, in RnnBatchStep()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomial.cpp128 const float* input_ptr_batch = inputData + b * class_size; in EvalFloat32() local
DRNN.cpp162 const T* input_ptr_batch = inputData + b * input_size; in RNNStep() local