projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b0396e3) | patch
[vulkan] Add log_softmax (#63193)
authorIvan Kobzarev <ivankobzarev@fb.com>
Tue, 17 Aug 2021 20:34:20 +0000 (13:34 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 17 Aug 2021 20:36:02 +0000 (13:36 -0700)
commit6621df9a6a1b362f18663fc8c180f3858b2414b3
tree62a81aac795046350c8283ad8d874da80898c549tree | snapshot
parentb0396e39f41da9f61c61ed8758b5e9505a370ebccommit | diff
[vulkan] Add log_softmax (#63193)

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

Test Plan: Imported from OSS

Reviewed By: SS-JIA

Differential Revision: D30291987

fbshipit-source-id: 89c6560274e5a841e5af249f6963b67ef6826f4c
aten/src/ATen/native/vulkan/glsl/log_softmax.glsl [new file with mode: 0644] blob
aten/src/ATen/native/vulkan/ops/Softmax.cpp diff | blob | history
aten/src/ATen/test/vulkan_api_test.cpp diff | blob | history
Domain: Machine Learning / ML Framework;