Skip to content

enable softmax op and add unit test#17703

Merged
hong19860320 merged 3 commits intoPaddlePaddle:incubate/litefrom
hong19860320:lite/softmax
May 29, 2019
Merged

enable softmax op and add unit test#17703
hong19860320 merged 3 commits intoPaddlePaddle:incubate/litefrom
hong19860320:lite/softmax

Conversation

@hong19860320
Copy link
Contributor

  1. enable softmax op
  2. add unit tests for softmax op

cmake ..
-DWITH_GPU=OFF
-DWITH_LITE=ON
-DLITE_WITH_ARM=ON
-DLITE_WITH_X86=OFF
-DLITE_WITH_CUDA=OFF
-DLITE_WITH_LIGHT_WEIGHT_FRAMEWORK=ON
-DWITH_TESTING=ON
-DWITH_MKLDNN=OFF

make test_softmax_op_lite
make test_softmax_compute_arm

@hong19860320 hong19860320 requested a review from Superjomn May 28, 2019 14:13
Copy link
Contributor

@tensor-tang tensor-tang left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

用pre-commit过一下格式

Copy link
Contributor

@Superjomn Superjomn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@hong19860320 hong19860320 merged commit e242f78 into PaddlePaddle:incubate/lite May 29, 2019
@hong19860320 hong19860320 deleted the lite/softmax branch May 29, 2019 06:53
Superjomn pushed a commit that referenced this pull request Jul 1, 2019
* enable softmax op and add unit test

* move softmax sub-functions to softmax.cc, and move basic math functions to funcs.h
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants