pytorch
5801431c - OpInfo Ref: addbmm (#61832)

Commit
3 years ago
OpInfo Ref: addbmm (#61832) Summary: See https://github.com/pytorch/pytorch/issues/54261. This PR: * Adds reference wrapper using NumPy for reference function of `addbmm` * Refines sample inputs (makes it more readable and avoids redundancy) cc: mruberry Pull Request resolved: https://github.com/pytorch/pytorch/pull/61832 Reviewed By: albanD Differential Revision: D29816024 Pulled By: mruberry fbshipit-source-id: e0fea6dc923504169a13bfaa258c61fbbc5fa9f4
Author
Parents
Loading