pytorch
f51a497f - [JIT] add builtin functions for (complex, Tensor) (#73286)

Commit
3 years ago
[JIT] add builtin functions for (complex, Tensor) (#73286) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/73286 Adds support in jit for op(complex, Tensor) for the following: * add (+) * mul (*) * eq (==) * ne (!=) * sub (-) * div (/) These already worked in eager mode, but didn't work in jit. Test Plan: Imported from OSS Reviewed By: george-qi Differential Revision: D34421794 Pulled By: davidberard98 fbshipit-source-id: 0aecec7d7da1dcf2bcceb6849d3049fe2b39076d (cherry picked from commit 375998d38d60ab97f7be8d95dd7bdc10bd1f6417)
Author
Committer
Parents
Loading