Skip to content

Commit 9043dd1

Browse files
committed
close all tensors
1 parent 0a7a43d commit 9043dd1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -353,6 +353,7 @@ public <T extends RealType<T> & NativeType<T>, R extends RealType<R> & NativeTyp
353353
for (int i = 0; i < resultPatchTensors.size(); i++) {
354354
try {
355355
rais.add(ImgLib2Builder.build(resultPatchTensors.get(i)));
356+
resultPatchTensors.get(i).close();
356357
} catch (IllegalArgumentException ex) {
357358
for (org.tensorflow.Tensor<?> tt : resultPatchTensors)
358359
tt.close();
@@ -435,6 +436,9 @@ protected List<String> inferenceFromShmas(List<String> inputs) throws IOExceptio
435436
ShmBuilder.build(resultPatchTensors.get(i), name, false);
436437
shmaNamesList.add(name);
437438
}
439+
for (org.tensorflow.Tensor<?> tt : resultPatchTensors) {
440+
tt.close();
441+
}
438442
return shmaNamesList;
439443
}
440444

0 commit comments

Comments
 (0)