Create Value with Tensor. (#9682)

Summary: #8364

Reviewed By: mergennachin

Differential Revision: D71914665

Co-authored-by: Anthony Shoumikhin <[email protected]>

Author: kirklandsign

extension/apple/ExecuTorch/Exported/ExecuTorchValue.h

@@ -47,13 +47,35 @@ __attribute__((deprecated("This API is experimental.")))
  */
 @property(nonatomic, readonly) ExecuTorchValueTag tag;
 
+/**
+ * The tensor value if the tag is ExecuTorchValueTagTensor.
+ *
+ * @return A Tensor instance or nil.
+ */
+@property(nullable, nonatomic, readonly) ExecuTorchTensor *tensorValue NS_SWIFT_NAME(tensor);
+
 /**
  * Returns YES if the value is of type None.
  *
  * @return A BOOL indicating whether the value is None.
  */
 @property(nonatomic, readonly) BOOL isNone;
 
+/**
+ * Returns YES if the value is a Tensor.
+ *
+ * @return A BOOL indicating whether the value is a Tensor.
+ */
+@property(nonatomic, readonly) BOOL isTensor;
+
+/**
+ * Creates an instance encapsulating a Tensor.
+ *
+ * @param value An ExecuTorchTensor instance.
+ * @return A new ExecuTorchValue instance with a tag of ExecuTorchValueTagTensor.
+ */
++ (instancetype)valueWithTensor:(ExecuTorchTensor *)value NS_SWIFT_NAME(init(_:));
+
 @end
 
 NS_ASSUME_NONNULL_END

extension/apple/ExecuTorch/Exported/ExecuTorchValue.mm

@@ -8,6 +8,8 @@
 
 #import "ExecuTorchValue.h"
 
+#import <executorch/runtime/platform/assert.h>
+
 @interface ExecuTorchValue ()
 
 - (instancetype)initWithTag:(ExecuTorchValueTag)tag
@@ -20,6 +22,11 @@ @implementation ExecuTorchValue {
   id _value;
 }
 
++ (instancetype)valueWithTensor:(ExecuTorchTensor *)value {
+  ET_CHECK(value);
+  return [[ExecuTorchValue alloc] initWithTag:ExecuTorchValueTagTensor value:value];
+}
+
 - (instancetype)init {
   return [self initWithTag:ExecuTorchValueTagNone value:nil];
 }
@@ -37,8 +44,16 @@ - (ExecuTorchValueTag)tag {
   return _tag;
 }
 
+- (nullable ExecuTorchTensor *)tensorValue {
+  return self.isTensor ? _value : nil;
+}
+
 - (BOOL)isNone {
   return _tag == ExecuTorchValueTagNone;
 }
 
+- (BOOL)isTensor {
+  return _tag == ExecuTorchValueTagTensor;
+}
+
 @end

extension/apple/ExecuTorch/__tests__/ValueTest.swift

@@ -15,4 +15,14 @@ class ValueTest: XCTestCase {
     let value = Value()
     XCTAssertTrue(value.isNone)
   }
+
+  func testTensor() {
+    var data: [Float] = [1.0, 2.0, 3.0, 4.0, 5.0, 6.0]
+    let tensor = data.withUnsafeMutableBytes {
+      Tensor(bytesNoCopy: $0.baseAddress!, shape: [2, 3], dataType: .float)
+    }
+    let value = Value(tensor)
+    XCTAssertTrue(value.isTensor)
+    XCTAssertEqual(value.tensor, tensor)
+  }
 }