Searched defs:reshaped_out (Results 1 – 1 of 1) sorted by relevance
2088 auto reshaped_out = t2_is_matrix ? out.reshape({folded_dim1, t2->sizes().back()}) in _matmul_impl() local2165 auto reshaped_out = out.reshape({expand_batch_product, n, p}); in _matmul_impl() local