Searched refs:aux_input_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 161 const float* input_to_output_weights_ptr, const float* aux_input_ptr, in LstmStepCalibration() argument
201 (aux_input_ptr == nullptr || in LstmStepCalibration()
202 tensor_utils::IsZeroVector(aux_input_ptr, n_batch * n_aux_input)); in LstmStepCalibration()
206 input_ptr, input_to_input_weights_ptr, aux_input_ptr, in LstmStepCalibration()
217 input_ptr, input_to_forget_weights_ptr, aux_input_ptr, in LstmStepCalibration()
226 CalculateLstmGateFloat(input_ptr, input_to_cell_weights_ptr, aux_input_ptr, in LstmStepCalibration()
241 input_ptr, input_to_output_weights_ptr, aux_input_ptr, in LstmStepCalibration()
341 const float* aux_input_ptr = nullptr; in EvalCalibration() local
343 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalCalibration()
352 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalCalibration()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 44 const float* input_to_output_weights_ptr, const float* aux_input_ptr, in LstmStepWithAuxInput() argument
130 if (aux_input_ptr != nullptr) { in LstmStepWithAuxInput()
133 aux_input_to_input_weights_ptr, n_cell, n_aux_input, aux_input_ptr, in LstmStepWithAuxInput()
138 aux_input_to_forget_weights_ptr, n_cell, n_aux_input, aux_input_ptr, in LstmStepWithAuxInput()
141 aux_input_to_cell_weights_ptr, n_cell, n_aux_input, aux_input_ptr, in LstmStepWithAuxInput()
144 aux_input_to_output_weights_ptr, n_cell, n_aux_input, aux_input_ptr, in LstmStepWithAuxInput()
410 const float* aux_input_ptr = nullptr; in EvalFloat() local
412 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalFloat()
421 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
460 const float* aux_input_ptr = nullptr; in EvalFloat() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 63 const float* aux_input_ptr) { in ComputeRowSums() argument
76 if (aux_input_ptr) { in ComputeRowSums()
812 const float* input_to_output_weights_ptr, const float* aux_input_ptr, in LstmStepFloat() argument
850 (aux_input_ptr == nullptr || in LstmStepFloat()
851 tensor_utils::IsZeroVector(aux_input_ptr, n_batch * n_aux_input)); in LstmStepFloat()
855 input_ptr, input_to_input_weights_ptr, aux_input_ptr, in LstmStepFloat()
865 input_ptr, input_to_forget_weights_ptr, aux_input_ptr, in LstmStepFloat()
873 CalculateLstmGateFloat(input_ptr, input_to_cell_weights_ptr, aux_input_ptr, in LstmStepFloat()
887 input_ptr, input_to_output_weights_ptr, aux_input_ptr, in LstmStepFloat()
989 float input_to_output_weights_scale, const float* aux_input_ptr, in LstmStepHybrid() argument
[all …]
|