aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/scatter_nd_op_cpu_impl_0.cc
Commit message (Collapse)AuthorAge
* Changes to scatter_nd opsGravatar Eugene Brevdo2016-11-08
| | | | | | | | | | * Rewrite CPU impl to be single-threaded and use vectorization; avoids race conditions. Removes use of the generator. * Remove scatter_nd_mul and scatter_nd_div to reduce binary size until we figure out a better way to reduce the templating pain * Modify scatter_nd to add for repeated indices as opposed to update (this is the appropriate gradient for gather_nd, for example) * Clean up docstrings. Change: 138452341
* Adding CPU kernels for tf.scatter_nd(), tf.scatter_nd_update(), ↵Gravatar A. Unique TensorFlower2016-11-02
tf.scatter_nd_add(), tf.scatter_nd_sub(), tf.scatter_nd_mul() and tf.scatter_nd_div() as well as gradient functions for tf.scatter_nd() and tf.gather_nd() Change: 138013328