Skip to content

Commit 6ad58b1

Browse files
committed
correct swithc
1 parent 67df3c1 commit 6ad58b1

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/main/java/io/bioimage/modelrunner/tensorflow/v1/shm/ShmBuilder.java

+5
Original file line numberDiff line numberDiff line change
@@ -68,14 +68,19 @@ public static void build(Tensor<?> tensor, String memoryName) throws IllegalArgu
6868
{
6969
case UINT8:
7070
buildFromTensorUByte((Tensor<UInt8>) tensor, memoryName);
71+
break;
7172
case INT32:
7273
buildFromTensorInt((Tensor<Integer>) tensor, memoryName);
74+
break;
7375
case FLOAT:
7476
buildFromTensorFloat((Tensor<Float>) tensor, memoryName);
77+
break;
7578
case DOUBLE:
7679
buildFromTensorDouble((Tensor<Double>) tensor, memoryName);
80+
break;
7781
case INT64:
7882
buildFromTensorLong((Tensor<Long>) tensor, memoryName);
83+
break;
7984
default:
8085
throw new IllegalArgumentException("Unsupported tensor type: " + tensor.dataType().name());
8186
}

0 commit comments

Comments
 (0)