projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 1abbee0) | patch
Optimize SoftmaxOp on CPU (#18635)
authorXiaomeng Yang <yangxm@fb.com>
Thu, 11 Apr 2019 01:45:57 +0000 (18:45 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 11 Apr 2019 01:52:15 +0000 (18:52 -0700)
commit821b5f138a987807032a2fd908fe10a5be5439d9
tree1a1a6c8cbf49939d166f949079a4d9efaf97a393tree | snapshot
parent1abbee0f8e088d2c99481a05672f07947916fd75commit | diff
Optimize SoftmaxOp on CPU (#18635)

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

Optimize SoftmaxOp on CPU

Reviewed By: houseroad

Differential Revision: D14689516

fbshipit-source-id: d2dcee2476d1a3a21f428e99bce9835f1d229d64
caffe2/operators/softmax_op.cc diff | blob | history
caffe2/operators/softmax_op.h diff | blob | history
caffe2/operators/softmax_shared.cc [deleted file] blob | history
caffe2/operators/softmax_shared.h [deleted file] blob | history
caffe2/operators/softmax_utils.cc [new file with mode: 0644] blob
caffe2/operators/softmax_utils.h [new file with mode: 0644] blob
caffe2/operators/softmax_with_loss_op.cc diff | blob | history
caffe2/operators/spatial_softmax_with_loss_op.cc diff | blob | history
modules/detectron/group_spatial_softmax_op.cc diff | blob | history
modules/detectron/softmax_focal_loss_op.cc diff | blob | history
Domain: Machine Learning / ML Framework;