Как заставить Tensorflow Eigen использовать библиотеку openBLAS для вызовов GEMM? - PullRequest
0 голосов
/ 23 марта 2020

Я пытаюсь связать OpenBLAS с Tensorflow Eigen. В основном я хочу переопределить вызов Eigen gemm для вызова OpenBLAS gemm, используя приведенную ниже команду, которая не работает.

Команда для построения TF. bazel build - c opt --copt = -mavx --copt = -mavx2 --copt = -mfma --copt = -mfpmath = оба --copt = -msse4.2 --copt = -DEIGEN_USE_BLAS --linkopt = -L --linkopt = -lopenblas // tenenflow / tools / pip_package: build_pip_package

Я ничего не нашел с документацией TF. Есть ли способ добиться того же?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...