diff --git a/js/react_native/cpp/TensorUtils.cpp b/js/react_native/cpp/TensorUtils.cpp index 79d270d883294..658e98491e17b 100644 --- a/js/react_native/cpp/TensorUtils.cpp +++ b/js/react_native/cpp/TensorUtils.cpp @@ -54,7 +54,7 @@ static const std::unordered_map {ONNX_TENSOR_ELEMENT_DATA_TYPE_INT8, "Int8Array"}, {ONNX_TENSOR_ELEMENT_DATA_TYPE_UINT16, "Uint16Array"}, {ONNX_TENSOR_ELEMENT_DATA_TYPE_INT16, "Int16Array"}, - {ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT16, "Float16Array"}, + {ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT16, "Uint16Array"}, {ONNX_TENSOR_ELEMENT_DATA_TYPE_STRING, "Array"}, {ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL, "Uint8Array"}, }; diff --git a/js/react_native/e2e/android/app/src/main/assets/test_types_float16.ort b/js/react_native/e2e/android/app/src/main/assets/test_types_float16.ort new file mode 100644 index 0000000000000..c691bc264d7ca Binary files /dev/null and b/js/react_native/e2e/android/app/src/main/assets/test_types_float16.ort differ diff --git a/js/react_native/e2e/ios/OnnxruntimeModuleExample.xcodeproj/project.pbxproj b/js/react_native/e2e/ios/OnnxruntimeModuleExample.xcodeproj/project.pbxproj index 70a5fcdd33cad..57e12e9bbdcef 100644 --- a/js/react_native/e2e/ios/OnnxruntimeModuleExample.xcodeproj/project.pbxproj +++ b/js/react_native/e2e/ios/OnnxruntimeModuleExample.xcodeproj/project.pbxproj @@ -15,6 +15,7 @@ 3ADD0A3E2EBB64D200761D6F /* ../src/test_types_int32.ort in Resources */ = {isa = PBXBuildFile; fileRef = 3ADD0A392EBB64D200761D6F /* ../src/test_types_int32.ort */; }; 3ADD0A3F2EBB64D200761D6F /* ../src/test_types_float.ort in Resources */ = {isa = PBXBuildFile; fileRef = 3ADD0A372EBB64D200761D6F /* ../src/test_types_float.ort */; }; 3ADD0A402EBB64D200761D6F /* ../src/test_types_uint8.ort in Resources */ = {isa = PBXBuildFile; fileRef = 3ADD0A3B2EBB64D200761D6F /* ../src/test_types_uint8.ort */; }; + 3ADD0A462EBB64D200761D6F /* ../src/test_types_float16.ort in Resources */ = {isa = PBXBuildFile; fileRef = 3ADD0A452EBB64D200761D6F /* ../src/test_types_float16.ort */; }; 3ADD0A422EBB677300761D6F /* test_types_double.onnx in Resources */ = {isa = PBXBuildFile; fileRef = 3ADD0A412EBB677300761D6F /* test_types_double.onnx */; }; 3ADD0A442EBB679A00761D6F /* test_types_bool.onnx in Resources */ = {isa = PBXBuildFile; fileRef = 3ADD0A432EBB679A00761D6F /* test_types_bool.onnx */; }; 81AB9BB82411601600AC10FF /* LaunchScreen.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = 81AB9BB72411601600AC10FF /* LaunchScreen.storyboard */; }; @@ -62,6 +63,7 @@ 3ADD0A392EBB64D200761D6F /* ../src/test_types_int32.ort */ = {isa = PBXFileReference; lastKnownFileType = file; path = ../src/test_types_int32.ort; sourceTree = ""; }; 3ADD0A3A2EBB64D200761D6F /* ../src/test_types_int64.ort */ = {isa = PBXFileReference; lastKnownFileType = file; path = ../src/test_types_int64.ort; sourceTree = ""; }; 3ADD0A3B2EBB64D200761D6F /* ../src/test_types_uint8.ort */ = {isa = PBXFileReference; lastKnownFileType = file; path = ../src/test_types_uint8.ort; sourceTree = ""; }; + 3ADD0A452EBB64D200761D6F /* ../src/test_types_float16.ort */ = {isa = PBXFileReference; lastKnownFileType = file; path = ../src/test_types_float16.ort; sourceTree = ""; }; 3ADD0A412EBB677300761D6F /* test_types_double.onnx */ = {isa = PBXFileReference; lastKnownFileType = file; path = ../src/test_types_double.onnx; sourceTree = ""; }; 3ADD0A432EBB679A00761D6F /* test_types_bool.onnx */ = {isa = PBXFileReference; lastKnownFileType = file; path = ../src/test_types_bool.onnx; sourceTree = ""; }; 81AB9BB72411601600AC10FF /* LaunchScreen.storyboard */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = file.storyboard; name = LaunchScreen.storyboard; path = OnnxruntimeModuleExample/LaunchScreen.storyboard; sourceTree = ""; }; @@ -149,6 +151,7 @@ 3ADD0A392EBB64D200761D6F /* ../src/test_types_int32.ort */, 3ADD0A3A2EBB64D200761D6F /* ../src/test_types_int64.ort */, 3ADD0A3B2EBB64D200761D6F /* ../src/test_types_uint8.ort */, + 3ADD0A452EBB64D200761D6F /* ../src/test_types_float16.ort */, DBA8BA86267293C4008CC55A /* mnist.ort */, DBBF7413263B8CCB00487C77 /* 3.jpg */, 13B07FAE1A68108700A75B9A /* OnnxruntimeModuleExample */, @@ -275,6 +278,7 @@ 3ADD0A3E2EBB64D200761D6F /* ../src/test_types_int32.ort in Resources */, 3ADD0A3F2EBB64D200761D6F /* ../src/test_types_float.ort in Resources */, 3ADD0A402EBB64D200761D6F /* ../src/test_types_uint8.ort in Resources */, + 3ADD0A462EBB64D200761D6F /* ../src/test_types_float16.ort in Resources */, E329E1162D3728940016B599 /* PrivacyInfo.xcprivacy in Resources */, 13B07FBF1A68108700A75B9A /* Images.xcassets in Resources */, ); diff --git a/js/react_native/e2e/src/BasicTypesTest.tsx b/js/react_native/e2e/src/BasicTypesTest.tsx index 0995f57a6872b..1bda629c6d4fd 100644 --- a/js/react_native/e2e/src/BasicTypesTest.tsx +++ b/js/react_native/e2e/src/BasicTypesTest.tsx @@ -39,6 +39,12 @@ const TEST_MODELS: TestModel[] = [ dataType: 'float32', description: 'Test float32 data type', }, + { + name: 'Float16', + asset: 'test_types_float16.ort', + dataType: 'float16', + description: 'Test float16 data type', + }, { name: 'Double', asset: 'test_types_double.onnx', @@ -211,6 +217,13 @@ export default class BasicTypesTest extends React.PureComponent<{}, State> { } return data; } + case 'float16': { + const data = new Uint16Array(size); + for (let i = 0; i < size; i++) { + data[i] = i; + } + return data; + } case 'float64': { const data = new Float64Array(size); for (let i = 0; i < size; i++) { diff --git a/js/react_native/e2e/src/test_types_float16.ort b/js/react_native/e2e/src/test_types_float16.ort new file mode 100644 index 0000000000000..c691bc264d7ca Binary files /dev/null and b/js/react_native/e2e/src/test_types_float16.ort differ