Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 327712789 Change-Id: Ia0afb258e4b8a48d398e3e73183d4baf5d79e7d3
This commit is contained in:
parent
44aa25ec46
commit
516d157980
@ -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.
|
||||||
|
Loading…
x
Reference in New Issue
Block a user