Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 327712789
Change-Id: Ia0afb258e4b8a48d398e3e73183d4baf5d79e7d3
diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go
index 8f0c1ef..746fd76 100644
--- a/tensorflow/go/op/wrappers.go
+++ b/tensorflow/go/op/wrappers.go
@@ -24328,9 +24328,28 @@
}
}
-// Returns the permuted vector/tensor in the destination data format given the
+// Permute input tensor from `src_format` to `dst_format`.
//
-// one in the source data format.
+// Input tensor must be a vector of size 4, or a 4x2 tensor.
+//
+// For example, with `src_format` of `NHWC`, `dst_format` of `NCHW`, and inputs:
+// ```
+// [1, 2, 3, 4]
+// ```
+// and
+// ```
+// [[1, 2, 3, 4],
+// [5, 6, 7, 8]]
+// ```
+// , the outputs will be (respectively):
+// ```
+// [1, 4, 2, 3]
+// ```
+// and
+// ```
+// [[1, 4, 2, 3],
+// [5, 8, 6, 7]]
+// ```
//
// Arguments:
// x: Vector of size 4 or Tensor of shape (4, 2) in source data format.
