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: 305332496
Change-Id: I9f03d4b13801e1985815c02fbdd5c8da82655e31
This commit is contained in:
A. Unique TensorFlower 2020-04-07 13:47:55 -07:00 committed by TensorFlower Gardener
parent 9327fd74f6
commit a21dd7457e
1 changed files with 54 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tensorflow/go/op/wrappers.go
View File

@ -29651,6 +29651,14 @@ func ResourceSparseApplyFtrlUseLocking(value bool) ResourceSparseApplyFtrlAttr {
}
}
// ResourceSparseApplyFtrlMultiplyLinearByLr sets the optional multiply_linear_by_lr attribute to value.
// If not specified, defaults to false
func ResourceSparseApplyFtrlMultiplyLinearByLr(value bool) ResourceSparseApplyFtrlAttr {
return func(m optionalAttr) {
m["multiply_linear_by_lr"] = value
}
}
// Update relevant entries in '*var' according to the Ftrl-proximal scheme.
//
// That is for rows we have grad for, we update var, accum and linear as follows:
@ -30664,6 +30672,14 @@ func ResourceApplyFtrlV2UseLocking(value bool) ResourceApplyFtrlV2Attr {
}
}
// ResourceApplyFtrlV2MultiplyLinearByLr sets the optional multiply_linear_by_lr attribute to value.
// If not specified, defaults to false
func ResourceApplyFtrlV2MultiplyLinearByLr(value bool) ResourceApplyFtrlV2Attr {
return func(m optionalAttr) {
m["multiply_linear_by_lr"] = value
}
}
// Update '*var' according to the Ftrl-proximal scheme.
//
// grad_with_shrinkage = grad + 2 * l2_shrinkage * var
@ -36054,6 +36070,14 @@ func ResourceApplyFtrlUseLocking(value bool) ResourceApplyFtrlAttr {
}
}
// ResourceApplyFtrlMultiplyLinearByLr sets the optional multiply_linear_by_lr attribute to value.
// If not specified, defaults to false
func ResourceApplyFtrlMultiplyLinearByLr(value bool) ResourceApplyFtrlAttr {
return func(m optionalAttr) {
m["multiply_linear_by_lr"] = value
}
}
// Update '*var' according to the Ftrl-proximal scheme.
//
// accum_new = accum + grad * grad
@ -42257,6 +42281,14 @@ func ResourceSparseApplyFtrlV2UseLocking(value bool) ResourceSparseApplyFtrlV2At
}
}
// ResourceSparseApplyFtrlV2MultiplyLinearByLr sets the optional multiply_linear_by_lr attribute to value.
// If not specified, defaults to false
func ResourceSparseApplyFtrlV2MultiplyLinearByLr(value bool) ResourceSparseApplyFtrlV2Attr {
return func(m optionalAttr) {
m["multiply_linear_by_lr"] = value
}
}
// Update relevant entries in '*var' according to the Ftrl-proximal scheme.
//
// That is for rows we have grad for, we update var, accum and linear as follows: