Searched refs:projection_weights_row_sums (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 48 int32_t* projection_weights_row_sums, int32_t* row_sums, int n_cell, in ComputeRowSums() argument
109 projection_weights_ptr, projection_weights_row_sums, n_output, n_cell); in ComputeRowSums()
438 int32_t* projection_weights_row_sums, bool* compute_row_sums, in CalculateLstmOutputHybrid() argument
474 projection_weights_row_sums, compute_row_sums, scratch2, context); in CalculateLstmOutputHybrid()
1058 int32_t* projection_weights_row_sums = nullptr; in LstmStepHybrid() local
1091 projection_weights_row_sums = recurrent_to_output_row_sums + n_cell; in LstmStepHybrid()
1101 projection_weights_row_sums, row_sums, n_cell, n_input, n_aux_input, in LstmStepHybrid()
1213 output_state_ptr, asymmetric_quantize_inputs, projection_weights_row_sums, in LstmStepHybrid()
|