Searched refs:cell_to_gate_weights (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 154 const float* cell_state, const float* cell_to_gate_weights, in CalculateLstmGateFloat() argument
160 const bool use_peephole = (cell_to_gate_weights != nullptr); in CalculateLstmGateFloat()
189 cell_to_gate_weights, n_cell, cell_state, n_batch, gate); in CalculateLstmGateFloat()
316 const float* cell_state, const int8_t* cell_to_gate_weights, in CalculateLstmGateHybrid() argument
335 const bool use_peephole = (cell_to_gate_weights != nullptr); in CalculateLstmGateHybrid()
396 tensor_utils::VectorScalarMultiply(cell_to_gate_weights, n_cell, in CalculateLstmGateHybrid()
498 const int16_t* cell_state, const int16_t* cell_to_gate_weights, in CalculateLstmGateInteger8x8_16() argument
514 const bool use_peephole = (cell_to_gate_weights != nullptr); in CalculateLstmGateInteger8x8_16()
535 cell_to_gate_weights, n_output, cell_state, n_batch, in CalculateLstmGateInteger8x8_16()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 44 const float* cell_state, const float* cell_to_gate_weights, in CalculateLstmGateFloat() argument
52 const bool use_peephole = (cell_to_gate_weights != nullptr); in CalculateLstmGateFloat()
81 cell_to_gate_weights, n_cell, cell_state, n_batch, gate); in CalculateLstmGateFloat()
|