fix: complete scalartype enum & move types to common.ts

Author: chmjkb

packages/react-native-executorch/src/modules/general/NewExecutorchModule.ts

@@ -1,27 +1,6 @@
 import { ResourceFetcher } from '../../utils/ResourceFetcher';
 import { ResourceSource } from '../../types/common';
-
-type TensorBuffer =
-  | Float32Array
-  | Float64Array
-  | Int8Array
-  | Int16Array
-  | Int32Array
-  | Uint8Array
-  | Uint16Array
-  | Uint32Array
-  | BigInt64Array
-  | BigUint64Array;
-
-enum ScalarType {
-  FLOAT16 = 1,
-}
-
-interface TensorPtr {
-  data: TensorBuffer;
-  shape: number[];
-  scalarType: ScalarType;
-}
+import { TensorPtr } from '../../types/common';
 
 export class NewExecutorchModule {
   nativeModule: any = null;
@@ -37,7 +16,7 @@ export class NewExecutorchModule {
     this.nativeModule = global.loadExecutorchModule(paths[0] || '');
   }
 
-  async forward(inputTensor: TensorPtr[]): Promise<void> {
+  async forward(inputTensor: TensorPtr[]): Promise<ArrayBuffer[]> {
     return await this.nativeModule.forward(inputTensor);
   }
 

packages/react-native-executorch/src/types/common.ts

@@ -15,3 +15,47 @@ export type ETInput =
   | BigInt64Array
   | Float32Array
   | Float64Array;
+
+export enum ScalarType {
+  BYTE = 0,
+  CHAR = 1,
+  SHORT = 2,
+  INT = 3,
+  LONG = 4,
+  HALF = 5,
+  FLOAT = 6,
+  DOUBLE = 7,
+  BOOL = 11,
+  QINT8 = 12,
+  QUINT8 = 13,
+  QINT32 = 14,
+  QUINT4X2 = 16,
+  QUINT2X4 = 17,
+  BITS16 = 22,
+  FLOAT8E5M2 = 23,
+  FLOAT8E4M3FN = 24,
+  FLOAT8E5M2FNUZ = 25,
+  FLOAT8E4M3FNUZ = 26,
+  UINT16 = 27,
+  UINT32 = 28,
+  UINT64 = 29,
+}
+
+export type TensorBuffer =
+  | ArrayBuffer
+  | Float32Array
+  | Float64Array
+  | Int8Array
+  | Int16Array
+  | Int32Array
+  | Uint8Array
+  | Uint16Array
+  | Uint32Array
+  | BigInt64Array
+  | BigUint64Array;
+
+export interface TensorPtr {
+  data: TensorBuffer;
+  shape: number[];
+  scalarType: ScalarType;
+}