网络/光纤>> mkldnn_rnn>> 返回
项目作者： yao-matrix

项目描述 ：
TF rnn ops w/ MKL-DNN kernel
高级语言： C++
项目地址: git://github.com/yao-matrix/mkldnn_rnn.git
创建时间： 2017-07-04T06:48:56Z
项目社区：https://github.com/yao-matrix/mkldnn_rnn
开源协议：
下载


How to build

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

How to run

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