Go: Update generated wrapper functions for TensorFlow ops.

PiperOrigin-RevId: 327712789
Change-Id: Ia0afb258e4b8a48d398e3e73183d4baf5d79e7d3
This commit is contained in:
A. Unique TensorFlower 2020-08-20 15:47:55 -07:00 committed by TensorFlower Gardener
parent 44aa25ec46
commit 516d157980

View File

@ -24328,9 +24328,28 @@ func DataFormatVecPermuteDstFormat(value string) DataFormatVecPermuteAttr {
} }
} }
// 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: // Arguments:
// x: Vector of size 4 or Tensor of shape (4, 2) in source data format. // x: Vector of size 4 or Tensor of shape (4, 2) in source data format.