projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 3bca429) | patch
Use 32 bit induction variable in gather expander
authorSanjoy Das <sanjoy@google.com>
Tue, 20 Mar 2018 18:13:48 +0000 (11:13 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 20 Mar 2018 18:18:12 +0000 (11:18 -0700)
commit1c4e42b39fd9ae2da14d7eb323bedc144a6e659b
treefa911d6aaf141acbbc1f7a8efc37c8c0165534a5tree | snapshot
parent3bca4298aacd9f89de2ac532bb7fedcdec1a5bb6commit | diff
Use 32 bit induction variable in gather expander

Right now this is unconditional (and we fail with Unimplemented() if a 32 bit
induction variable is not large enough), but eventually we may want to be
smarter about this.

PiperOrigin-RevId: 189773581
tensorflow/compiler/xla/service/BUILD diff | blob | history
tensorflow/compiler/xla/service/gather_expander.cc diff | blob | history
tensorflow/compiler/xla/service/gather_expander_test.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/service/while_util.cc diff | blob | history
tensorflow/compiler/xla/service/while_util.h diff | blob | history
tensorflow/compiler/xla/tests/BUILD diff | blob | history
tensorflow/compiler/xla/tests/gather_operation_test.cc diff | blob | history
tensorflow/compiler/xla/util.h diff | blob | history
Domain: Machine Learning / ML Framework;