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.

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

Add Fused Batch Normalization #437

Merged
merged 2 commits into from May 22, 2019

Conversation

Projects
None yet
2 participants
@TE-TakuyaNarihira
Copy link
Contributor

commented May 22, 2019

The PR sony/nnabla-ext-cuda#154 adds CUDA implementation.
A fused operation of batch normalization + add2 + activation is added. This reduces memory usage and execution time especially in CUDA backend.

Depends on #436.

@TE-TakuyaNarihira TE-TakuyaNarihira force-pushed the feature/20190522-fused-bn-nhwc branch from 2fcc021 to db91a15 May 22, 2019

@TE-AkioHayakawa TE-AkioHayakawa merged commit 863e298 into master May 22, 2019

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.