Searched refs:n_bw_cell (Results 1 – 1 of 1) sorted by relevance
454 const int n_bw_cell = bw_input_to_output_weights->dims->data[0]; in Prepare() local478 n_bw_cell); in Prepare()600 n_bw_cell)); in Prepare()628 TF_LITE_ENSURE_EQ(context, NumElements(bw_cell_state), n_batch * n_bw_cell); in Prepare()650 bw_scratch_buffer_size->data[1] = n_bw_cell * 3; in Prepare()653 bw_scratch_buffer_size->data[1] = n_bw_cell * 4; in Prepare()835 int n_cell = std::max(n_fw_cell, n_bw_cell); in Prepare()927 bw_row_sums_rows += ceil(static_cast<float>(n_bw_output) / n_bw_cell); in Prepare()936 int bw_row_sums_dims[2] = {bw_row_sums_rows, n_bw_cell}; in Prepare()