@@ -415,7 +415,7 @@ private List<String> encodeInputs(List<Tensor<?>> inputTensors) throws FileAlrea
415
415
HashMap <String , Object > map = new HashMap <String , Object >();
416
416
map .put (NAME_KEY , tt .getName ());
417
417
map .put (SHAPE_KEY , tt .getShape ());
418
- map .put (DTYPE_KEY , CommonUtils .getDataType (tt .getData ()));
418
+ map .put (DTYPE_KEY , CommonUtils .getDataTypeFromRAI (tt .getData ()));
419
419
map .put (IS_INPUT_KEY , true );
420
420
map .put (MEM_NAME_KEY , shma .getName ());
421
421
encodedInputTensors .add (gson .toJson (map ));
@@ -433,7 +433,7 @@ private List<String> encodeOutputs(List<Tensor<?>> outputTensors) throws FileAlr
433
433
map .put (IS_INPUT_KEY , false );
434
434
if (!tt .isEmpty ()) {
435
435
map .put (SHAPE_KEY , tt .getShape ());
436
- map .put (DTYPE_KEY , CommonUtils .getDataType (tt .getData ()));
436
+ map .put (DTYPE_KEY , CommonUtils .getDataTypeFromRAI (tt .getData ()));
437
437
SharedMemoryArray shma = SharedMemoryArray .createSHMAFromRAI (tt .getData (), false , true );
438
438
shmaOutputList .add (shma );
439
439
map .put (MEM_NAME_KEY , shma .getName ());
0 commit comments