saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-11 17:48:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[js/rn] Support create boolean tensor (#17052)

Browse source 
### Description
<!-- Describe your changes. -->

For some use case need to create boolean tensor.

I've tested on [this
project](https://github.com/hans00/react-native-transformers-example)

### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

Add handle `ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL`

And it required #15556 (It seems not include in latest release
(v1.15.1))
This commit is contained in:
Hans 2023-09-14 13:02:27 +08:00 committed by GitHub
parent 32f5658abb
commit ad369a1fad
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 33 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
js/react_native/android/src/androidTest/java/ai/onnxruntime/reactnative/TensorHelperTest.java
View file

@ -238,6 +238,34 @@ public class TensorHelperTest {
outputTensor.close();
}
@Test
public void createInputTensor_bool() throws Exception {
OnnxTensor outputTensor = OnnxTensor.createTensor(ortEnvironment, new boolean[] {false, true});
JavaOnlyMap inputTensorMap = new JavaOnlyMap();
JavaOnlyArray dims = new JavaOnlyArray();
dims.pushInt(2);
inputTensorMap.putArray("dims", dims);
inputTensorMap.putString("type", TensorHelper.JsTensorTypeBool);
ByteBuffer dataByteBuffer = ByteBuffer.allocate(2);
dataByteBuffer.put((byte)0);
dataByteBuffer.put((byte)1);
inputTensorMap.putMap("data", blobModule.testCreateData(dataByteBuffer.array()));
OnnxTensor inputTensor = TensorHelper.createInputTensor(blobModule, inputTensorMap, ortEnvironment);
Assert.assertEquals(inputTensor.getInfo().onnxType, TensorInfo.OnnxTensorType.ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL);
Assert.assertEquals(outputTensor.getInfo().onnxType, TensorInfo.OnnxTensorType.ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL);
Assert.assertEquals(inputTensor.toString(), outputTensor.toString());
Assert.assertArrayEquals(inputTensor.getByteBuffer().array(), outputTensor.getByteBuffer().array());
inputTensor.close();
outputTensor.close();
}
@Test
public void createOutputTensor_bool() throws Exception {
MockitoSession mockSession = mockitoSession().mockStatic(Arguments.class).startMocking();

6
js/react_native/android/src/main/java/ai/onnxruntime/reactnative/TensorHelper.java
View file

@ -174,7 +174,11 @@ public class TensorHelper {
tensor = OnnxTensor.createTensor(ortEnvironment, buffer, dims, OnnxJavaType.UINT8);
break;
}
case ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL:
case ONNX_TENSOR_ELEMENT_DATA_TYPE_BOOL: {
ByteBuffer buffer = values;
tensor = OnnxTensor.createTensor(ortEnvironment, buffer, dims, OnnxJavaType.BOOL);
break;
}
case ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT16:
case ONNX_TENSOR_ELEMENT_DATA_TYPE_UINT16:
case ONNX_TENSOR_ELEMENT_DATA_TYPE_UINT32: