---

yaml
---
r: 311263
b: refs/heads/tensorflow-merge
c: c6a98b1
h: refs/heads/master
i:
  311261: 8a85446
  311259: 8481a06
  311255: 4bbf959
  311247: 96f08ce
  311231: 28636ac

Author: LaurentMazare

[refs]

@@ -1379,7 +1379,7 @@ refs/heads/chase-my-tail: 8bb91443a9e81bbfac92a2621a0af887a1da8dbf
 refs/heads/consider-outer-alternatives: 708bac749ec60a22a79e2eefbe734f9488a7370d
 refs/heads/revert-25740-oops-i-linked-it-again: fdd41aeb682fc488572bdc1cf71b2ff6997ba576
 refs/heads/swift-5.1-branch-06-12-2019: e63b7b2d3b93c48232d386099d0ec525d21d8f8d
-refs/heads/tensorflow-merge: cbad7aac3a6d6edb321919ae39f6353429d7f820
+refs/heads/tensorflow-merge: c6a98b121f7a0bfb0aac89d328f1ba3b8162200d
 refs/heads/update-checkout-sha-info: 5832743c5c2a842976c42a508a4c6dcceefb0aef
 refs/tags/swift-5.1-DEVELOPMENT-SNAPSHOT-2019-06-12-a: 228f0448d9bb909aacbba4afcb7c600a405d15da
 refs/tags/swift-5.1-DEVELOPMENT-SNAPSHOT-2019-06-14-a: 922861a77b5fc2bf46bc917da70ceb15eef76836

branches/tensorflow-merge/stdlib/public/TensorFlow/Tensor.swift

@@ -536,10 +536,12 @@ public extension Tensor where Scalar : Numeric {
   @_inlineable @inline(__always)
   init(oneHotAtIndices indices: Tensor<Int32>, depth: Int32,
        onValue: Scalar = 1, offValue: Scalar = 0, axis: Int = -1) {
-    self.init(
-      handle: #tfop("OneHot", indices, Tensor<Int32>(depth),
-                    Tensor(onValue), Tensor(offValue),
-                    axis: axis, T: Scalar.self, TI: Int32.self)
+    self = Raw.oneHot(
+      indices: indices,
+      depth: Tensor<Int32>(depth),
+      onValue: Tensor(onValue),
+      offValue: Tensor(offValue),
+      axis: Int64(axis)
     )
   }
 }