Searched refs:n_idx (Results 1 – 2 of 2) sorted by relevance
219 int n_idx = ndims_out - 2; in CreateMatMulParams() local221 n = rhs_dims[n_idx]; in CreateMatMulParams()222 std::swap(rhs_dims[k_idx], rhs_dims[n_idx]); in CreateMatMulParams()224 rhs_strides[n_idx] = 1; in CreateMatMulParams()
13 // negative index 'n_idx' as: n_idx = p_idx-num_dims