pytorch
141bfbef - [iOS GPU] Add tanh and clamp to support GAN (#61383)

Commit
3 years ago
[iOS GPU] Add tanh and clamp to support GAN (#61383) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61383 Since we already have the support for hardtanh, it's easy to add support for clamp. GPU is 40% ish faster. ghstack-source-id: 133113272 Test Plan: - CI - buck test pp-macos Reviewed By: dhruvbird Differential Revision: D29572933 fbshipit-source-id: d22ec09e18d02456440f552067c9a8aea9a1a8ab
Author
Parents
Loading