Searched refs:cell_state_scale (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 572 int32_t cell_state_scale, const int16_t* input_gate, in UpdateLstmCellInteger() argument
585 30 + cell_state_scale, scratch); in UpdateLstmCellInteger()
588 30 + cell_state_scale, scratch); in UpdateLstmCellInteger()
618 int32_t cell_state_scale, const int16_t* output_gate, in CalculateLstmOutputInteger8x8_16() argument
626 tensor_utils::ApplyTanh(15 + cell_state_scale, cell_state, n_batch, n_cell, in CalculateLstmOutputInteger8x8_16()
1364 int32_t cell_state_scale, int32_t input_variance_guard, in LstmStepInteger8x8_16()
1446 UpdateLstmCellInteger(n_batch, n_cell, cell_state_ptr, cell_state_scale, in LstmStepInteger8x8_16()
1464 n_batch, n_cell, n_output, cell_state_ptr, cell_state_scale, in LstmStepInteger8x8_16()
|