Searched refs:input_to_cell_weights_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 52 const int8_t* input_to_cell_weights_ptr, in ComputeRowSums() argument
71 tensor_utils::ReductionSumVector(input_to_cell_weights_ptr, in ComputeRowSums()
811 const float* input_to_cell_weights_ptr, in LstmStepFloat() argument
873 CalculateLstmGateFloat(input_ptr, input_to_cell_weights_ptr, aux_input_ptr, in LstmStepFloat()
984 const int8_t* input_to_cell_weights_ptr, in LstmStepHybrid() argument
1103 input_to_cell_weights_ptr, input_to_output_weights_ptr, in LstmStepHybrid()
1173 quantized_input_ptr, input_sf, input_zp, input_to_cell_weights_ptr, in LstmStepHybrid()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 160 const float* input_to_cell_weights_ptr, in LstmStepCalibration() argument
226 CalculateLstmGateFloat(input_ptr, input_to_cell_weights_ptr, aux_input_ptr, in LstmStepCalibration()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 43 const float* input_to_cell_weights_ptr, in LstmStepWithAuxInput() argument
108 tensor_utils::MatrixBatchVectorMultiplyAccumulate(input_to_cell_weights_ptr, in LstmStepWithAuxInput()
|