pytorch

fb9fc42a - optimize BatchMatmulOp (#18612)

optimize BatchMatmulOp (#18612) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18612 optimize BatchMatmulOp Reviewed By: houseroad Differential Revision: D14681665 fbshipit-source-id: cf5ea4909ace58fd44fe6fa634531102ac84e851