diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-09 19:46:01 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-09 19:48:25 -0700 |
commit | a356b2128a9bdbc33eceeff4b058f4d5d2e97738 (patch) | |
tree | 4c92d6173740c3208b27fbf28a6c975a75dc4f8f /tensorflow/go | |
parent | e3e0af4bd9b1d7a4628a5a4d6901a2d8529cfda5 (diff) |
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 192227995
Diffstat (limited to 'tensorflow/go')
-rw-r--r-- | tensorflow/go/op/wrappers.go | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go index 3d261c9d0a..09da8c1892 100644 --- a/tensorflow/go/op/wrappers.go +++ b/tensorflow/go/op/wrappers.go @@ -13882,6 +13882,14 @@ func DecodeCSVNaValue(value string) DecodeCSVAttr { } } +// DecodeCSVSelectCols sets the optional select_cols attribute to value. +// If not specified, defaults to <> +func DecodeCSVSelectCols(value []int64) DecodeCSVAttr { + return func(m optionalAttr) { + m["select_cols"] = value + } +} + // Convert CSV records to tensors. Each column maps to one tensor. // // RFC 4180 format is expected for the CSV records. |