Searched refs:ndims_rhs (Results 1 – 1 of 1) sorted by relevance
94 const auto ndims_rhs = rhs.dims(); in Compute() local108 auto rhs_rows = rhs.dim_size(ndims_rhs - 2); in Compute()109 auto rhs_cols = rhs.dim_size(ndims_rhs - 1); in Compute()181 const auto ndims_rhs = rhs_shape.dims(); in CreateMatMulParams() local195 if (ndims_rhs < ndims_out) { in CreateMatMulParams()