aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorA. Unique TensorFlower <gardener@tensorflow.org>2024-05-09 15:45:12 -0700
committerTensorFlower Gardener <gardener@tensorflow.org>2024-05-09 16:33:23 -0700
commit25b937da3c7ff64724bb27c7f53234b76d8223ca (patch)
treea0116bd26e048f29cf11715a55e4fddc1e491983
parentcb904f490f618e1182a2f27eb166209509f055b3 (diff)
downloadtensorflow-25b937da3c7ff64724bb27c7f53234b76d8223ca.tar.gz
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 632292573
-rw-r--r--tensorflow/go/op/wrappers.go4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go
index ed921329d2f..c5fd9bf5196 100644
--- a/tensorflow/go/op/wrappers.go
+++ b/tensorflow/go/op/wrappers.go
@@ -19563,6 +19563,10 @@ func GatherV2BatchDims(value int64) GatherV2Attr {
// On GPU, if an out of bound index is found, a 0 is stored in the
// corresponding output value.
//
+// Note that on TPU, if any dimension of `params` is of size 0 then the output will
+// be the expected shape filled with zeros. On CPU and GPU an error will be
+// returned.
+//
// See also `tf.batch_gather` and `tf.gather_nd`.
//
// Arguments: