Searched refs:is_aux_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
70 if (!is_aux_input_all_zeros) { in CalculateLstmGateFloat()
200 const bool is_aux_input_all_zeros = in LstmStepCalibration() local
212 is_input_all_zeros, is_aux_input_all_zeros, logger, in LstmStepCalibration()
223 is_aux_input_all_zeros, logger, intermediate_tensor_indexes[1], in LstmStepCalibration()
233 is_input_all_zeros, is_aux_input_all_zeros, logger, in LstmStepCalibration()
247 is_aux_input_all_zeros, logger, intermediate_tensor_indexes[3], 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
178 if (!is_aux_input_all_zeros) { in CalculateLstmGateFloat()
327 const bool is_input_all_zeros, const bool is_aux_input_all_zeros, in CalculateLstmGateHybrid()
367 if (!is_aux_input_all_zeros) { in CalculateLstmGateHybrid()
849 const bool is_aux_input_all_zeros = in LstmStepFloat() local
861 is_input_all_zeros, is_aux_input_all_zeros); in LstmStepFloat()
871 is_aux_input_all_zeros); in LstmStepFloat()
880 is_input_all_zeros, is_aux_input_all_zeros); in LstmStepFloat()
893 is_aux_input_all_zeros); in LstmStepFloat()
1116 const bool is_aux_input_all_zeros = in LstmStepHybrid() local
[all …]
|