From b370544b8f7787730ed742a4fe1d3d4ceba1926e Mon Sep 17 00:00:00 2001
From: Alex Black <blacka101@gmail.com>
Date: Fri, 16 Aug 2019 11:32:02 +1000
Subject: [PATCH] Small fix in TensorflowConversion class (#121)

Signed-off-by: Alex Black <blacka101@gmail.com>
---
 .../org/nd4j/tensorflow/conversion/TensorflowConversion.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java b/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java
index c5ae88198..b47cd30d1 100644
--- a/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java
+++ b/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java
@@ -296,7 +296,7 @@ public class TensorflowConversion {
      * @return
      */
     public static String defaultDeviceForThread() {
-        Integer deviceForThread = Nd4j.getAffinityManager().getDeviceForThread(Thread.currentThread());
+        Integer deviceForThread = Nd4j.getAffinityManager().getDeviceForCurrentThread();
         String deviceName = null;
         //gpu
         if(Nd4j.getBackend().getClass().getName().contains("JCublasBackend")) {