tree 3e4b9b892a4eac27ac22bdcb33ee01da0f3ea5c1
parent e38d865abefe343e0ae79e9abc75b96475734454
author Robert David <lrdx@google.com> 1623977399 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1623978051 -0700

Add pattern to replace
  (float)OneHot($X, $N, $on_val, $off_val, $axis)
with
  OneHot($X, $N, (float)$on_val, (float)$off_val, $axis)

PiperOrigin-RevId: 380084267
Change-Id: I6b0eb4ec81e592ec19935c5c531736701569f561