diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-24 11:46:17 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-24 11:49:15 -0700 |
commit | 9992042548ff268ac97ac3ebf1c584d380b0c106 (patch) | |
tree | 03a69a4a8c8ddd436643c7a640b8f362381789c0 /tensorflow/go | |
parent | c2b1eebe7e256dda88beb91c7fa7662e01d12f9b (diff) |
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 194120868
Diffstat (limited to 'tensorflow/go')
-rw-r--r-- | tensorflow/go/op/wrappers.go | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go index d038846c4f..4d91f2b68e 100644 --- a/tensorflow/go/op/wrappers.go +++ b/tensorflow/go/op/wrappers.go @@ -9602,6 +9602,14 @@ func ResourceApplyAdagradUseLocking(value bool) ResourceApplyAdagradAttr { } } +// ResourceApplyAdagradUpdateSlots sets the optional update_slots attribute to value. +// If not specified, defaults to true +func ResourceApplyAdagradUpdateSlots(value bool) ResourceApplyAdagradAttr { + return func(m optionalAttr) { + m["update_slots"] = value + } +} + // Update '*var' according to the adagrad scheme. // // accum += grad * grad @@ -10676,6 +10684,14 @@ func ResourceSparseApplyAdagradUseLocking(value bool) ResourceSparseApplyAdagrad } } +// ResourceSparseApplyAdagradUpdateSlots sets the optional update_slots attribute to value. +// If not specified, defaults to true +func ResourceSparseApplyAdagradUpdateSlots(value bool) ResourceSparseApplyAdagradAttr { + return func(m optionalAttr) { + m["update_slots"] = value + } +} + // Update relevant entries in '*var' and '*accum' according to the adagrad scheme. // // That is for rows we have grad for, we update var and accum as follows: |