Searched refs:unreduced_dims (Results 1 – 1 of 1) sorted by relevance
49 const int unreduced_dims = (dims + !reduce_first_axis_) / 2; in permutation() local51 for (int i = 0; i < unreduced_dims; i++) { in permutation()54 for (int i = unreduced_dims; i < dims; i++) { in permutation()55 perm[i] = 2 * (i - unreduced_dims) + !reduce_first_axis_; in permutation()