Searched refs:num_row_sums (Results 1 – 2 of 2) sorted by relevance
294 const int num_row_sums = has_aux_input ? 3 : 2; in Prepare() local303 int fw_row_sums_dims[2] = {num_row_sums, fw_num_units}; in Prepare()319 int bw_row_sums_dims[2] = {num_row_sums, bw_num_units}; in Prepare()
1061 int num_row_sums = use_cifg ? 6 : 8; in LstmStepHybrid() local1063 num_row_sums += use_cifg ? 3 : 4; in LstmStepHybrid()1066 num_row_sums += ceil(static_cast<float>(n_output) / n_cell); in LstmStepHybrid()1068 TF_LITE_ASSERT(row_sums_size == num_row_sums); in LstmStepHybrid()