diff options
| author |  Adria Puigdomenech <adriap@google.com> | 2018-08-21 02:51:23 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 02:55:55 -0700 |
| commit | fd833e87f150a8963de906593f0b0071cd962fcb (patch) | |
| tree | 16ef036d1bf780154a2ac1d3094cdb16897beb54 /tensorflow/python/BUILD | |
| parent | 9e310b6253d2ec6e57559b77b64faee787385604 (diff) | |
Add `batch_scatter_update`, analogous to `batch_gather`.
This operation computes:
ref[i_1, ..., i_n, indices[i_1, ..., i_n, j]] = updates[i_1, ..., i_n, j]
That is, it assumes that `ref`, `indices` and `updates` have a series of leading dimensions that are the same for all of them, and the updates are performed on the last dimension of indices.
PiperOrigin-RevId: 209566652
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 68b3722326..07c74ec71d 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -2780,11 +2780,13 @@ py_library( srcs = ["ops/state_ops.py"], srcs_version = "PY2AND3", deps = [ + ":array_ops", ":framework_ops", + ":math_ops_gen", ":resource_variable_ops_gen", ":state_ops_gen", ":tensor_shape", - "//tensorflow/python/eager:context", + ":util", ], ) |