We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7e2106c commit 833d388Copy full SHA for 833d388
extension/android/executorch_android/src/main/java/org/pytorch/executorch/Tensor.java
@@ -714,7 +714,7 @@ private static Tensor nativeNewTensor(
714
tensor = new Tensor_int8(data, shape);
715
} else if (DType.HALF.jniCode == dtype) {
716
tensor = new Tensor_raw_data_16b(data, shape, DType.HALF);
717
- } else if (DType.HALF.jniCode == dtype) {
+ } else if (DType.BFLOAT16.jniCode == dtype) {
718
tensor = new Tensor_raw_data_16b(data, shape, DType.BFLOAT16);
719
} else {
720
throw new IllegalArgumentException("Unknown Tensor dtype");
0 commit comments