Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small fix in TensorflowConversion class (#121) 

Signed-off-by: Alex Black <blacka101@gmail.com>
master
Alex Black 2019-08-16 11:32:02 +10:00 committed by GitHub
parent 0a3e78b540
commit b370544b8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java
View File

@ -296,7 +296,7 @@ public class TensorflowConversion {
* @return * @return
*/ */
public static String defaultDeviceForThread() { public static String defaultDeviceForThread() {
Integer deviceForThread = Nd4j.getAffinityManager().getDeviceForThread(Thread.currentThread()); Integer deviceForThread = Nd4j.getAffinityManager().getDeviceForCurrentThread();
String deviceName = null; String deviceName = null;
//gpu //gpu
if(Nd4j.getBackend().getClass().getName().contains("JCublasBackend")) { if(Nd4j.getBackend().getClass().getName().contains("JCublasBackend")) {