aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/go
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-04-24 11:46:17 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-24 11:49:15 -0700
commit9992042548ff268ac97ac3ebf1c584d380b0c106 (patch)
tree03a69a4a8c8ddd436643c7a640b8f362381789c0 /tensorflow/go
parentc2b1eebe7e256dda88beb91c7fa7662e01d12f9b (diff)
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 194120868
Diffstat (limited to 'tensorflow/go')
-rw-r--r--tensorflow/go/op/wrappers.go16
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: