Join GitHub today
GitHub is home to over 36 million developers working together to host and review code, manage projects, and build software together.Sign up
[WIP] DL4J MKL-DNN Support #7151
Adds MKL-DNN support for DL4J layers.
MKL-DNN support for:
Note that conv3d is already using conv3dnew op, hence already supports MKL-DNN by default.
Previous performance issues with MKL-DNN have been resolved.
LRN is implemented but is disabled on account of this: #7272
LSTM is not implemneted here - we don't yet have MKL-DNN support at libnd4j blockLSTM op (or similar) level.