Add MKL based tensorflow back-end

This commit is contained in:
Edwin Jakobs
2020-10-21 14:02:51 +02:00
parent 2e40843715
commit 8f4c3d746d
5 changed files with 17 additions and 1 deletions

View File

@@ -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()) {

View File

@@ -0,0 +1,4 @@
dependencies {
runtimeOnly "org.bytedeco:mkl-dnn:$mklDnnVersion:linux-x86_64"
runtimeOnly "org.tensorflow:tensorflow-core-api:$tensorflowVersion:linux-x86_64-mkl"
}

View File

@@ -0,0 +1,4 @@
dependencies {
runtimeOnly "org.bytedeco:mkl-dnn:$mklDnnVersion:macosx-x86_64"
runtimeOnly "org.tensorflow:tensorflow-core-api:$tensorflowVersion:macosx-x86_64-mkl"
}

View File

@@ -0,0 +1,4 @@
dependencies {
runtimeOnly "org.bytedeco:mkl-dnn:$mklDnnVersion:windows-x86_64"
runtimeOnly "org.tensorflow:tensorflow-core-api:$tensorflowVersion:windows-x86_64"
}

View File

@@ -40,9 +40,12 @@ include 'openrndr-demos',
'orx-syphon',
'orx-temporal-blur',
'orx-tensorflow',
'orx-tensorflow-mkl-natives-linux-x64',
'orx-tensorflow-mkl-natives-macos',
'orx-tensorflow-mkl-natives-windows',
'orx-tensorflow-natives-linux-x64',
'orx-tensorflow-natives-windows',
'orx-tensorflow-natives-macos',
'orx-tensorflow-natives-windows',
'orx-timer',
'orx-time-operators',
'orx-kinect-common',