pytorch
736deefc - [torch][te] aten::type_as is unary, not binary (#48085)

Commit
4 years ago
[torch][te] aten::type_as is unary, not binary (#48085) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/48085 We were treating it as a binary operator, which implies shape broadcasting, even though the second arg is thrown away aside from the type. Treating it as a unary is the proper approach. ghstack-source-id: 116873680 Test Plan: new unit test Reviewed By: ZolotukhinM Differential Revision: D25017585 fbshipit-source-id: 0cfa89683c9bfd4fbb132617c74b47b268d7f368
Author
Parents
Loading