[TF] Change APIs to use Int instead of Int32. (#84)

Friend PR: swiftlang/swift#24012

Author: dan-zheng

Sources/DeepLearning/Initializers.swift

@@ -61,7 +61,7 @@ public extension Tensor where Scalar: BinaryFloatingPoint {
                                 Int64.random(in: Int64.min..<Int64.max))
     ) {
         self = Raw.statelessRandomUniform(
-          shape: Tensor<Int32>((0..<shape.rank).map{shape[$0]}),
+          shape: Tensor<Int32>((0..<shape.rank).map { Int32(shape[$0]) }),
           seed: Tensor<Int64>([seed.0, seed.1])
         )
     }
@@ -79,7 +79,7 @@ public extension Tensor where Scalar: BinaryFloatingPoint {
                                 Int64.random(in: Int64.min..<Int64.max))
     ) {
         self = Raw.statelessRandomNormal(
-            shape: Tensor<Int32>((0..<shape.rank).map{shape[$0]}),
+            shape: Tensor<Int32>((0..<shape.rank).map { Int32(shape[$0]) }),
             seed: Tensor<Int64>([seed.0, seed.1])
         )
     }