projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: d2eda7f) | patch
Port `log_softmax` to structured kernel (#57374)
authorMeghan Lele <meghanl@fb.com>
Thu, 12 Aug 2021 04:01:28 +0000 (21:01 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 12 Aug 2021 04:02:48 +0000 (21:02 -0700)
commitba603594fd8e74165773cc3656d6aa388e11ac60
tree7fa18c555dfd3b0ef7520f7680040feeda838a6etree | snapshot
parentd2eda7f2f3d3121068aaa9fa8410d9b1fc34fe53commit | diff
Port `log_softmax` to structured kernel (#57374)

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

Test Plan: Imported from OSS

Reviewed By: saketh-are

Differential Revision: D30240243

Pulled By: SplitInfinity

fbshipit-source-id: de6617c75d16e26d607a884c25b8752b7b561737
aten/src/ATen/native/SoftMax.cpp diff | blob | history
aten/src/ATen/native/cpu/SoftMaxKernel.cpp diff | blob | history
aten/src/ATen/native/cpu/SoftmaxKernel.h diff | blob | history
aten/src/ATen/native/cuda/SoftMax.cu diff | blob | history
aten/src/ATen/native/native_functions.yaml diff | blob | history
aten/src/ATen/native/sparse/SoftMax.cpp diff | blob | history
aten/src/ATen/native/sparse/cuda/SoftMax.cu diff | blob | history
Domain: Machine Learning / ML Framework;