Skip to content

yao-matrix/mkldnn_rnn

Repository files navigation

How to build

1. put this git in ./tensorflow/contrib folder
2. cd <tensorflow-root-path>
3. $ git am ./tensorflow/contrib/mkldnn_rnn/0001-enable-mkldnn_rnn-in-tensorflow-codebase.patch -3
4. $ cp ./tensorflow/contrib/mkldnn_rnn/build.sh ./
5. $ ./configure, pls enable mkl while configuration
6. $ . ./build.sh

How to run

  1. functionality test
  $ python ./python/kernel_tests/mkldnn_rnn_ops_test.py
  1. benchmark test
  $ python ./python/kernel_tests/mkldnn_rnn_ops_benchmark.py

About

TF rnn ops w/ MKL-DNN kernel

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •