projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 9fe297f) | patch
Use the new gather HLO in the bridge when lowering TF gather ops; NFC
authorSanjoy Das <sanjoy@google.com>
Wed, 18 Apr 2018 20:47:17 +0000 (13:47 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 18 Apr 2018 20:52:45 +0000 (13:52 -0700)
commitf17311fa8d2df24e56deaab743cdf1ec5e12c692
tree890f9f00d118d76a03f12b4c8ec7cd13b46fd9a1tree | snapshot
parent9fe297ffa8133309fe548df3a0208d0ff9305a66commit | diff
Use the new gather HLO in the bridge when lowering TF gather ops; NFC

After gather expansion this should boil down to a while loop very similar to
what we emit from the bridge today.

PiperOrigin-RevId: 193410095
tensorflow/compiler/tf2xla/kernels/gather_op.cc diff | blob | history
tensorflow/compiler/tf2xla/kernels/gather_op_helpers.h diff | blob | history
Domain: Machine Learning / ML Framework;