Skip to content

Commit f31bcca

Browse files
authored
[Java] add missing optional tensor list functions (#6640)
Update [ghstack-poisoned]
1 parent 63017e4 commit f31bcca

File tree

1 file changed

+7
-1
lines changed
  • extension/android/src/main/java/org/pytorch/executorch

1 file changed

+7
-1
lines changed

extension/android/src/main/java/org/pytorch/executorch/EValue.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ public class EValue {
6161
"ListInt",
6262
"ListTensor",
6363
"ListScalar",
64-
"ListOptionalScalar",
64+
"ListOptionalTensor",
6565
};
6666

6767
@DoNotStrip private final int mTypeCode;
@@ -267,6 +267,12 @@ public Tensor[] toTensorList() {
267267
return (Tensor[]) mData;
268268
}
269269

270+
@DoNotStrip
271+
public Optional<Tensor>[] toOptionalTensorList() {
272+
preconditionType(TYPE_CODE_LIST_OPTIONAL_TENSOR, mTypeCode);
273+
return (Optional<Tensor>[]) mData;
274+
}
275+
270276
private void preconditionType(int typeCodeExpected, int typeCode) {
271277
if (typeCode != typeCodeExpected) {
272278
throw new IllegalStateException(

0 commit comments

Comments
 (0)