Skip to content

Commit 29fcdec

Browse files
committed
add missing optional tensor list functions
ghstack-source-id: aa8c449 ghstack-comment-id: 2455710950 Pull Request resolved: #6640
1 parent 97a4600 commit 29fcdec

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)