Searched refs:ndims_out (Results 1 – 1 of 1) sorted by relevance
182 const auto ndims_out = out_shape.dims(); in CreateMatMulParams() local192 if (ndims_lhs < ndims_out) { in CreateMatMulParams()195 if (ndims_rhs < ndims_out) { in CreateMatMulParams()208 int m_idx = ndims_out - 1; in CreateMatMulParams()209 int k_idx = ndims_out - 2; in CreateMatMulParams()218 int k_idx = ndims_out - 1; in CreateMatMulParams()219 int n_idx = ndims_out - 2; in CreateMatMulParams()