[PR-2715] Feedback

deepjavalibrary · Jul 25, 2023 · f0dae63 · f0dae63
1 parent 1cbf2ca
commit f0dae63
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArrayEx.java b/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArrayEx.java
@@ -695,10 +695,10 @@ public NDList multiBoxPrior(
             List<Float> offsets,
             boolean clip) {
 
-        NDManager ndManager = array.getManager().getParentManager();
+        NDManager ndManager = array.getManager();
 
-        Float stepX = steps.get(1);
-        Float stepY = steps.get(0);
+        float stepX = steps.get(1);
+        float stepY = steps.get(0);
         int numSizes = sizes.size();
         int numRatios = ratios.size();
         int count = 0;