Searched refs:ndims_lhs (Results 1 – 1 of 1) sorted by relevance
93 const auto ndims_lhs = lhs.dims(); in Compute() local106 auto lhs_rows = lhs.dim_size(ndims_lhs - 2); in Compute()107 auto lhs_cols = lhs.dim_size(ndims_lhs - 1); in Compute()180 const auto ndims_lhs = lhs_shape.dims(); in CreateMatMulParams() local192 if (ndims_lhs < ndims_out) { in CreateMatMulParams()