projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 35413a1) | patch
[NNC] Add Softplus operator (#64589)
authorAnimesh Jain <anijain@fb.com>
Wed, 8 Sep 2021 17:48:09 +0000 (10:48 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 8 Sep 2021 17:49:58 +0000 (10:49 -0700)
commit18d24bb537bbe567fdb9f7b268c51478987319b8
tree916bb84a9506d08480368942dbaf761784401ce6tree | snapshot
parent35413a16f7433b345ba1dc9c460cbe9c2d975762commit | diff
[NNC] Add Softplus operator (#64589)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64589

Adding softplus operator lowering for NNC. Enabling element wise fusion as well.

Test Plan: Added a test in test_jit_fuser.py

Reviewed By: bertmaher

Differential Revision: D30736449

fbshipit-source-id: 6c5fc3bceb5cef2322ecd4449f827e4af018ea93
test/test_jit_fuser_te.py diff | blob | history
torch/csrc/jit/passes/tensorexpr_fuser.cpp diff | blob | history
torch/csrc/jit/tensorexpr/kernel.cpp diff | blob | history
Domain: Machine Learning / ML Framework;