Searched refs:input_to_input_weights_ptr (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 81 const int8_t* input_to_input_weights_ptr, in ComputeRowSums() argument
97 tensor_utils::ReductionSumVector(input_to_input_weights_ptr, in ComputeRowSums()
852 const float* input_ptr, const float* input_to_input_weights_ptr, in LstmStepFloat() argument
882 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepFloat()
900 CalculateLstmGateFloat(input_ptr, input_to_input_weights_ptr, aux_input_ptr, in LstmStepFloat()
1034 const float* input_ptr, const int8_t* input_to_input_weights_ptr, in LstmStepHybrid() argument
1095 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepHybrid()
1158 n_output, input_to_input_weights_ptr, input_to_forget_weights_ptr, in LstmStepHybrid()
1196 quantized_input_ptr, input_sf, input_zp, input_to_input_weights_ptr, in LstmStepHybrid()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 41 const float* input_ptr, const float* input_to_input_weights_ptr, in LstmStepWithAuxInput() argument
72 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepWithAuxInput()
101 input_to_input_weights_ptr, n_cell, n_input, input_ptr, n_batch, in LstmStepWithAuxInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 159 const float* input_ptr, const float* input_to_input_weights_ptr, in LstmStepCalibration() argument
190 const bool use_cifg = (input_to_input_weights_ptr == nullptr); in LstmStepCalibration()
207 input_ptr, input_to_input_weights_ptr, aux_input_ptr, in LstmStepCalibration()
|