Searched refs:bw_row_sums_dims (Results 1 – 2 of 2) sorted by relevance
319 int bw_row_sums_dims[2] = {num_row_sums, bw_num_units}; in Prepare() local320 if (!TfLiteIntArrayEqualsArray(bw_row_sums->dims, 2, bw_row_sums_dims)) { in Prepare()322 bw_row_sums_size->data[0] = bw_row_sums_dims[0]; in Prepare()323 bw_row_sums_size->data[1] = bw_row_sums_dims[1]; in Prepare()
936 int bw_row_sums_dims[2] = {bw_row_sums_rows, n_bw_cell}; in Prepare() local937 if (!TfLiteIntArrayEqualsArray(bw_row_sums->dims, 2, bw_row_sums_dims)) { in Prepare()939 bw_row_sums_size->data[0] = bw_row_sums_dims[0]; in Prepare()940 bw_row_sums_size->data[1] = bw_row_sums_dims[1]; in Prepare()