Decomposition of bmm and mm for dot product (#106593)
Summary: Decomposition of bmm and mm for dot product
Test Plan: sandcastle, and Bert
Differential Revision: D48055141
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106593
Approved by: https://github.com/jansel