Searched refs:recurrent_to_gate_weights (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 153 const float* output_state, const float* recurrent_to_gate_weights, in CalculateLstmGateFloat() argument
185 recurrent_to_gate_weights, n_cell, n_output, output_state, n_batch, gate); in CalculateLstmGateFloat()
311 const int32_t* output_state_zp, const int8_t* recurrent_to_gate_weights, in CalculateLstmGateHybrid() argument
383 recurrent_to_gate_weights, recurrent_to_gate_weights_ledger, n_cell, in CalculateLstmGateHybrid()
387 recurrent_to_gate_weights, n_cell, n_output, output_state, in CalculateLstmGateHybrid()
493 const int8_t* output_state, const int8_t* recurrent_to_gate_weights, in CalculateLstmGateInteger8x8_16() argument
529 output_state, recurrent_to_gate_bias, recurrent_to_gate_weights, in CalculateLstmGateInteger8x8_16()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 43 const float* output_state, const float* recurrent_to_gate_weights, in CalculateLstmGateFloat() argument
77 recurrent_to_gate_weights, n_cell, n_output, output_state, n_batch, gate); in CalculateLstmGateFloat()
|