Searched refs:aux_input_to_gate_weights (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 42 const float* aux_input, const float* aux_input_to_gate_weights, in CalculateLstmGateFloat() argument
71 tensor_utils::MatrixBatchVectorMultiplyAccumulate(aux_input_to_gate_weights, in CalculateLstmGateFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 152 const float* aux_input, const float* aux_input_to_gate_weights, in CalculateLstmGateFloat() argument
179 tensor_utils::MatrixBatchVectorMultiplyAccumulate(aux_input_to_gate_weights, in CalculateLstmGateFloat()
306 const int32_t* aux_input_zp, const int8_t* aux_input_to_gate_weights, in CalculateLstmGateHybrid() argument
369 aux_input_to_gate_weights, n_cell, n_aux_input, aux_input, in CalculateLstmGateHybrid()
|