correct swithc

bioimage-io · Oct 2, 2024 · 6ad58b1 · 6ad58b1
1 parent 67df3c1
commit 6ad58b1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java
@@ -68,14 +68,19 @@ public static void build(Tensor<?> tensor, String memoryName) throws IllegalArgu
         {
             case UINT8:
             	buildFromTensorUByte((Tensor<UInt8>) tensor, memoryName);
+            	break;
             case INT32:
             	buildFromTensorInt((Tensor<Integer>) tensor, memoryName);
+            	break;
             case FLOAT:
             	buildFromTensorFloat((Tensor<Float>) tensor, memoryName);
+            	break;
             case DOUBLE:
             	buildFromTensorDouble((Tensor<Double>) tensor, memoryName);
+            	break;
             case INT64:
             	buildFromTensorLong((Tensor<Long>) tensor, memoryName);
+            	break;
             default:
                 throw new IllegalArgumentException("Unsupported tensor type: " + tensor.dataType().name());
         }