File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
src/main/java/io/bioimage/modelrunner/pytorch/shm Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -61,18 +61,24 @@ private ShmBuilder()
61
61
*/
62
62
public static void build (NDArray tensor , String memoryName ) throws IllegalArgumentException , IOException
63
63
{
64
+ System .out .println (tensor .getDataType ().asNumpy ());
64
65
switch (tensor .getDataType ())
65
66
{
66
67
case UINT8 :
67
68
buildFromTensorUByte (tensor , memoryName );
69
+ break ;
68
70
case INT32 :
69
71
buildFromTensorInt (tensor , memoryName );
72
+ break ;
70
73
case FLOAT32 :
71
74
buildFromTensorFloat (tensor , memoryName );
75
+ break ;
72
76
case FLOAT64 :
73
77
buildFromTensorDouble (tensor , memoryName );
78
+ break ;
74
79
case INT64 :
75
80
buildFromTensorLong (tensor , memoryName );
81
+ break ;
76
82
default :
77
83
throw new IllegalArgumentException ("Unsupported tensor type: " + tensor .getDataType ().asNumpy ());
78
84
}
You can’t perform that action at this time.
0 commit comments