Searched refs:k_idx (Results 1 – 1 of 1) sorted by relevance
209 int k_idx = ndims_out - 2; in CreateMatMulParams() local211 k = lhs_dims[k_idx]; in CreateMatMulParams()212 std::swap(lhs_dims[m_idx], lhs_dims[k_idx]); in CreateMatMulParams()214 lhs_strides[k_idx] = 1; in CreateMatMulParams()218 int k_idx = ndims_out - 1; in CreateMatMulParams() local220 k = rhs_dims[k_idx]; in CreateMatMulParams()222 std::swap(rhs_dims[k_idx], rhs_dims[n_idx]); in CreateMatMulParams()223 rhs_strides[k_idx] = k; in CreateMatMulParams()