Home
last modified time | relevance | path

Searched defs:input_weights_ptr (Results 1 – 4 of 4) sorted by relevance

/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/external/tensorflow/tensorflow/lite/kernels/
H A Dbasic_rnn.cc218 const float* input_weights_ptr = GetTensorData<float>(input_weights); in EvalFloat() local
251 const int8_t* input_weights_ptr = GetTensorData<int8_t>(input_weights); in EvalHybrid() local
H A Dunidirectional_sequence_rnn.cc225 const float* input_weights_ptr = GetTensorData<float>(input_weights); in EvalFloat() local
289 const int8_t* input_weights_ptr = GetTensorData<int8_t>(input_weights); in EvalHybrid() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DRNN.cpp171 const T* input_weights_ptr = weightsData; in RNNStep() local