experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Go: Update generated wrapper functions for TensorFlow ops.

Browse Source 
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
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tensorflow/go/op/wrappers.go
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:
// x: Vector of size 4 or Tensor of shape (4, 2) in source data format.