2626
2727import  java .io .IOException ;
2828import  java .nio .ByteBuffer ;
29+ import  java .nio .ByteOrder ;
2930import  java .util .Arrays ;
3031
3132import  org .tensorflow .types .TFloat32 ;
@@ -102,9 +103,9 @@ private static void buildFromTensorUByte(TUint8 tensor, String memoryName) throw
102103        SharedMemoryArray  shma  = SharedMemoryArray .readOrCreate (memoryName , arrayShape , new  UnsignedByteType (), false , true );
103104        ByteBuffer  buff  = shma .getDataBufferNoHeader ();
104105        byte [] flat  = new  byte [buff .capacity ()];
105-         ByteBuffer  buff2  = ByteBuffer .wrap (flat );
106+         ByteBuffer  buff2  = ByteBuffer .wrap (flat ). order ( ByteOrder . LITTLE_ENDIAN ) ;
106107        tensor .asRawTensor ().data ().read (flat , 0 , buff .capacity ());
107-         buff  =  buff2 ;
108+         buff . put ( buff2 ) ;
108109        if  (PlatformDetection .isWindows ()) shma .close ();
109110    }
110111
@@ -118,9 +119,9 @@ private static void buildFromTensorInt(TInt32 tensor, String memoryName) throws
118119        SharedMemoryArray  shma  = SharedMemoryArray .readOrCreate (memoryName , arrayShape , new  IntType (), false , true );
119120        ByteBuffer  buff  = shma .getDataBufferNoHeader ();
120121        byte [] flat  = new  byte [buff .capacity ()];
121-         ByteBuffer  buff2  = ByteBuffer .wrap (flat );
122+         ByteBuffer  buff2  = ByteBuffer .wrap (flat ). order ( ByteOrder . LITTLE_ENDIAN ) ;
122123        tensor .asRawTensor ().data ().read (flat , 0 , buff .capacity ());
123-         buff  =  buff2 ;
124+         buff . put ( buff2 ) ;
124125        if  (PlatformDetection .isWindows ()) shma .close ();
125126    }
126127
@@ -134,9 +135,9 @@ private static void buildFromTensorFloat(TFloat32 tensor, String memoryName) thr
134135        SharedMemoryArray  shma  = SharedMemoryArray .readOrCreate (memoryName , arrayShape , new  FloatType (), false , true );
135136        ByteBuffer  buff  = shma .getDataBufferNoHeader ();
136137        byte [] flat  = new  byte [buff .capacity ()];
137-         ByteBuffer  buff2  = ByteBuffer .wrap (flat );
138+         ByteBuffer  buff2  = ByteBuffer .wrap (flat ). order ( ByteOrder . LITTLE_ENDIAN ) ;
138139        tensor .asRawTensor ().data ().read (flat , 0 , buff .capacity ());
139-         buff  =  buff2 ;
140+         buff . put ( buff2 ) ;
140141        if  (PlatformDetection .isWindows ()) shma .close ();
141142    }
142143
@@ -150,9 +151,9 @@ private static void buildFromTensorDouble(TFloat64 tensor, String memoryName) th
150151        SharedMemoryArray  shma  = SharedMemoryArray .readOrCreate (memoryName , arrayShape , new  DoubleType (), false , true );
151152        ByteBuffer  buff  = shma .getDataBufferNoHeader ();
152153        byte [] flat  = new  byte [buff .capacity ()];
153-         ByteBuffer  buff2  = ByteBuffer .wrap (flat );
154+         ByteBuffer  buff2  = ByteBuffer .wrap (flat ). order ( ByteOrder . LITTLE_ENDIAN ) ;
154155        tensor .asRawTensor ().data ().read (flat , 0 , buff .capacity ());
155-         buff  =  buff2 ;
156+         buff . put ( buff2 ) ;
156157        if  (PlatformDetection .isWindows ()) shma .close ();
157158    }
158159
@@ -167,9 +168,9 @@ private static void buildFromTensorLong(TInt64 tensor, String memoryName) throws
167168        SharedMemoryArray  shma  = SharedMemoryArray .readOrCreate (memoryName , arrayShape , new  LongType (), false , true );
168169        ByteBuffer  buff  = shma .getDataBufferNoHeader ();
169170        byte [] flat  = new  byte [buff .capacity ()];
170-         ByteBuffer  buff2  = ByteBuffer .wrap (flat );
171+         ByteBuffer  buff2  = ByteBuffer .wrap (flat ). order ( ByteOrder . LITTLE_ENDIAN ) ;
171172        tensor .asRawTensor ().data ().read (flat , 0 , buff .capacity ());
172-         buff  =  buff2 ;
173+         buff . put ( buff2 ) ;
173174        if  (PlatformDetection .isWindows ()) shma .close ();
174175    }
175176}
0 commit comments