Commit message (Collapse) | Author | Age | |
---|---|---|---|
* | Changes to scatter_nd ops | Eugene Brevdo | 2016-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(), ↵ | A. Unique TensorFlower | 2016-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 |