30ecfa3abe3c4146c29e1b6ee9a3279efa0833d7 |
|
13-Feb-2018 |
Peter Hawkins <phawkins@google.com> |
[TF:XLA] Implement ScatterNd. Add a helper method for performing scatter operations. Share it between ScatterNd and UnsortedSegmentSum implementations. In passing, add support for negative indices to the UnsortedSegmentSum implementation. Added helper methods for creating XLA while loops. Use the new helper in both Gather and Scatter ops. PiperOrigin-RevId: 185469229
/external/tensorflow/tensorflow/compiler/tf2xla/lib/scatter.h
|