Searched refs:n_aux_input (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 49 int n_input, int n_aux_input, int n_output, in ComputeRowSums() argument
80 n_aux_input); in ComputeRowSums()
84 n_aux_input); in ComputeRowSums()
87 n_aux_input); in ComputeRowSums()
90 n_aux_input); in ComputeRowSums()
156 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat() argument
180 n_cell, n_aux_input, in CalculateLstmGateFloat()
321 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateHybrid() argument
369 aux_input_to_gate_weights, n_cell, n_aux_input, aux_input, in CalculateLstmGateHybrid()
832 int n_aux_input, int n_output, int output_batch_leading_dim, in LstmStepFloat()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 46 const int n_batch, const int n_input, const int n_aux_input, in CalculateLstmGateFloat() argument
72 n_cell, n_aux_input, in CalculateLstmGateFloat()
181 int n_aux_input, int n_output, int output_batch_leading_dim, in LstmStepCalibration()
202 tensor_utils::IsZeroVector(aux_input_ptr, n_batch * n_aux_input)); in LstmStepCalibration()
210 input_gate_bias_ptr, n_batch, n_input, n_aux_input, n_output, n_cell, in LstmStepCalibration()
221 forget_gate_bias_ptr, n_batch, n_input, n_aux_input, n_output, n_cell, in LstmStepCalibration()
231 n_batch, n_input, n_aux_input, n_output, n_cell, in LstmStepCalibration()
245 output_gate_bias_ptr, n_batch, n_input, n_aux_input, n_output, n_cell, in LstmStepCalibration()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 64 int n_aux_input, int n_output, int output_batch_leading_dim, 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()
|