Add MKL based tensorflow back-end

2020-10-21 14:02:51 +02:00
parent 2e40843715
commit 8f4c3d746d
5 changed files with 17 additions and 1 deletions
--- a/build.gradle
+++ b/build.gradle
@@ -25,6 +25,7 @@ project.ext {
    gsonVersion = "2.8.6"
    antlrVersion = "4.8-1"
    tensorflowVersion = "0.2.0"
+    mklDnnVersion = "0.21.5-1.5.4"
 }

 switch (org.gradle.internal.os.OperatingSystem.current()) {