Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Introducing MKL to softmax for inference #14437

Merged
merged 3 commits into from Nov 20, 2018
Merged

Introducing MKL to softmax for inference #14437

merged 3 commits into from Nov 20, 2018
+53 −38

Commits on Nov 16, 2018

  1. Squashing MKL based softmax for inference 

    test=develop

- Added profiling to softmax functors

- MKL based softmax inference op

- Fix to softmax compuation via MKL

- cleaning

- Cosmetic fixes to softmax MKL

- Fix to ON_INFER lack of propagation
    @jczaja
    jczaja committed Nov 16, 2018
    Configuration menu
    Copy the full SHA
    513bb6c View commit details
    Browse the repository at this point in the history

  2. - Fix to GPU 

    test=develop
    @jczaja
    jczaja committed Nov 16, 2018
    Configuration menu
    Copy the full SHA
    be80bb4 View commit details
    Browse the repository at this point in the history

Commits on Nov 18, 2018

  1. - Removing partial specialization of sotmax for inference for GPU 

    test=develop
    @jczaja
    jczaja committed Nov 18, 2018
    Configuration menu
    Copy the full SHA
    9b0eae3 View commit details
    Browse the repository at this point in the history