Searched refs:projection_weights_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 62 const int8_t* projection_weights_ptr, bool use_cifg, in ComputeRowSums() argument
107 if (projection_weights_ptr != nullptr) { in ComputeRowSums()
109 projection_weights_ptr, projection_weights_row_sums, n_output, n_cell); in ComputeRowSums()
830 const float* projection_weights_ptr, const float* projection_bias_ptr, in LstmStepFloat() argument
897 projection_weights_ptr, projection_bias_ptr, in LstmStepFloat()
1021 const int8_t* projection_weights_ptr, in LstmStepHybrid()
1065 if (projection_weights_ptr != nullptr) { in LstmStepHybrid()
1090 if (projection_weights_ptr != nullptr) { in LstmStepHybrid()
1108 projection_weights_ptr, use_cifg, aux_input_ptr); in LstmStepHybrid()
1211 params->activation, projection_weights_ptr, projection_weights_ledger_ptr, in LstmStepHybrid()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 62 const float* projection_weights_ptr, const float* projection_bias_ptr, in LstmStepWithAuxInput() argument
292 const bool use_projection_weight = (projection_weights_ptr != nullptr); in LstmStepWithAuxInput()
311 projection_weights_ptr, n_output, n_cell, in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 179 const float* projection_weights_ptr, const float* projection_bias_ptr, in LstmStepCalibration() argument
252 params->activation, projection_weights_ptr, projection_bias_ptr, in LstmStepCalibration()
|