Switched from using 'Raw.add' to using 'Raw.addV2'. (#433)

Author: eaplatanios

Sources/TensorFlow/Core/Tensor.swift

@@ -510,7 +510,7 @@ extension Tensor: AdditiveArithmetic where Scalar: Numeric {
     @inlinable
     @differentiable(vjp: _vjpAdd(lhs:rhs:) where Scalar: TensorFlowFloatingPoint)
     public static func + (lhs: Tensor, rhs: Tensor) -> Tensor {
-        Raw.add(lhs, rhs)
+        Raw.addV2(lhs, rhs)
     }
 
     /// Subtracts one tensor from another and produces their difference.

Tests/TensorFlowTests/LazyTensorExplicitTraceTests.swift

@@ -34,7 +34,7 @@ final class LazyTensorExplicitTraceTests: XCTestCase {
         XCTAssertEqual(trace.description,
             """
             lazyTrace_2(%0: float) -> (%1) {
-              %1 = Add[T: float](%0, %0)
+              %1 = AddV2[T: float](%0, %0)
             }
             """)
         let outputs = runTrace(trace: trace, input: Tensor<Float>(10.0))
@@ -55,7 +55,7 @@ final class LazyTensorExplicitTraceTests: XCTestCase {
               %2 = Const[dtype: int32, value: 4]()
               %3 = Mul[T: int32](%1, %2)
               %4 = Const[dtype: float, value: 3.0]()
-              %5 = Add[T: float](%0, %4)
+              %5 = AddV2[T: float](%0, %4)
             }
             """)
         let outputs = runTrace(
@@ -78,7 +78,7 @@ final class LazyTensorExplicitTraceTests: XCTestCase {
             """
             lazyTrace_4(%0: float) -> (%3) {
               %1 = Const[dtype: float, value: 10.0]()
-              %2 = Add[T: float](%1, %1)
+              %2 = AddV2[T: float](%1, %1)
               %3 = Mul[T: float](%0, %2)
             }
             """)
@@ -127,7 +127,7 @@ final class LazyTensorExplicitTraceTests: XCTestCase {
             """
             lazyTrace_3(%0: float) -> (%2) {
               %1 = Const[dtype: float, value: 9.0]()
-              %2 = Add[T: float](%1, %0)
+              %2 = AddV2[T: float](%1, %0)
             }
             """)
         let outputs = runTrace(trace: trace, input: Tensor<Float>(4.0))

Tests/TensorFlowTests/LazyTensorTFFunctionBuilderTests.swift

@@ -67,10 +67,10 @@ final class LazyTensorTFFunctionBuilderTests : XCTestCase {
             tfFunction(w, "sequence")!.description,
             """
 
-            sequence(placeholder_0:float, placeholder_1:float, placeholder_2:float) -> (add_4:float) {
+            sequence(placeholder_0:float, placeholder_1:float, placeholder_2:float) -> (addv2_4:float) {
               Mul_3 = Mul[T=float](placeholder_1, placeholder_2)
-              Add_4 = Add[T=float](placeholder_0, Mul_3:z:0)
-              return add_4 = Add_4:z:0
+              AddV2_4 = AddV2[T=float](placeholder_0, Mul_3:z:0)
+              return addv2_4 = AddV2_4:z:0
             }
 
             """)

Tests/TensorFlowTests/LazyTensorTraceTests.swift

@@ -40,7 +40,7 @@ final class LazyTensorTraceTests: XCTestCase {
               %1 = Const[dtype: float, value: 2.0]()
               %2 = Const[dtype: float, value: 3.0]()
               %3 = Mul[T: float](%1, %2)
-              %4 = Add[T: float](%0, %3)
+              %4 = AddV2[T: float](%0, %3)
             }
             """)
     }
@@ -60,9 +60,9 @@ final class LazyTensorTraceTests: XCTestCase {
             lazyTrace_8() -> (%4, %5, %7) {
               %0 = Const[dtype: float, value: 10.0]()
               %1 = Const[dtype: float, value: 2.0]()
-              %2 = Add[T: float](%0, %1)
+              %2 = AddV2[T: float](%0, %1)
               %3 = Const[dtype: float, value: 3.0]()
-              %4 = Add[T: float](%2, %3)
+              %4 = AddV2[T: float](%2, %3)
               %5 = Mul[T: float](%4, %3)
               %6 = Sub[T: float](%4, %3)
               %7 = Div[T: float](%5, %6)
@@ -76,9 +76,9 @@ final class LazyTensorTraceTests: XCTestCase {
             lazyTrace_6() -> (%4, %5) {
               %0 = Const[dtype: float, value: 10.0]()
               %1 = Const[dtype: float, value: 2.0]()
-              %2 = Add[T: float](%0, %1)
+              %2 = AddV2[T: float](%0, %1)
               %3 = Const[dtype: float, value: 3.0]()
-              %4 = Add[T: float](%2, %3)
+              %4 = AddV2[T: float](%2, %3)
               %5 = Mul[T: float](%4, %3)
             }
             """)
@@ -97,10 +97,10 @@ final class LazyTensorTraceTests: XCTestCase {
             lazyTrace_6() -> (%2, %5) {
               %0 = Const[dtype: float, value: 1.0]()
               %1 = Const[dtype: float, value: 2.0]()
-              %2 = Add[T: float](%0, %1)
+              %2 = AddV2[T: float](%0, %1)
               %3 = Const[dtype: float, value: 3.0]()
               %4 = Const[dtype: float, value: 4.0]()
-              %5 = Add[T: float](%3, %4)
+              %5 = AddV2[T: float](%3, %4)
             }
             """)
     }
@@ -116,7 +116,7 @@ final class LazyTensorTraceTests: XCTestCase {
             """
             lazyTrace_2() -> (%1) {
               %0 = Const[dtype: float, value: 5.0]()
-              %1 = Add[T: float](%0, %0)
+              %1 = AddV2[T: float](%0, %0)
             }
             """)
         let mul = addOrMul(/*useAdd:*/false, a)
@@ -182,7 +182,7 @@ final class LazyTensorTraceTests: XCTestCase {
             lazyTrace_3() -> (%2) {
               %0 = Const[dtype: float, value: 1.0]()
               %1 = Const[dtype: float, value: 2.0]()
-              %2 = Add[T: float](%0, %1)
+              %2 = AddV2[T: float](%0, %1)
             }
             """)
         XCTAssertEqual(y.scalarized(), 3.0)