Searched refs:is_input_all_zeros (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 49 const bool is_input_all_zeros, const bool is_aux_input_all_zeros, in CalculateLstmGateFloat() argument
64 if (!is_input_all_zeros) { in CalculateLstmGateFloat()
198 const bool is_input_all_zeros = in LstmStepCalibration() local
212 is_input_all_zeros, is_aux_input_all_zeros, logger, in LstmStepCalibration()
222 /*activation=*/kTfLiteActSigmoid, forget_gate_scratch, is_input_all_zeros, in LstmStepCalibration()
233 is_input_all_zeros, is_aux_input_all_zeros, logger, in LstmStepCalibration()
246 /*activation=*/kTfLiteActSigmoid, output_gate_scratch, is_input_all_zeros, in LstmStepCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 159 const bool is_input_all_zeros, const bool is_aux_input_all_zeros) { in CalculateLstmGateFloat() argument
172 if (!is_input_all_zeros) { in CalculateLstmGateFloat()
327 const bool is_input_all_zeros, const bool is_aux_input_all_zeros, in CalculateLstmGateHybrid()
347 if (!is_input_all_zeros) { in CalculateLstmGateHybrid()
847 const bool is_input_all_zeros = in LstmStepFloat() local
861 is_input_all_zeros, is_aux_input_all_zeros); in LstmStepFloat()
870 /*activation=*/kTfLiteActSigmoid, forget_gate_scratch, is_input_all_zeros, in LstmStepFloat()
880 is_input_all_zeros, is_aux_input_all_zeros); in LstmStepFloat()
892 /*activation=*/kTfLiteActSigmoid, output_gate_scratch, is_input_all_zeros, in LstmStepFloat()
1114 const bool is_input_all_zeros = in LstmStepHybrid() local
[all …]
|