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: 308353110
Change-Id: Ib648a4149c35212609b313d1e48da7db6b9e5788
This commit is contained in:
A. Unique TensorFlower 2020-04-24 16:56:09 -07:00 committed by TensorFlower Gardener
parent a4e22aa1c4
commit 7550e571ac
1 changed files with 26 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6915,9 +6915,7 @@ func GetSessionHandle(scope *Scope, value tf.Output) (handle tf.Output) {
return op.Output(0) return op.Output(0)
} }
// Copy a tensor setting everything outside a central band in each innermost matrix // Copy a tensor setting everything outside a central band in each innermost matrix to zero.
//
// to zero.
// //
// The `band` part is computed as follows: // The `band` part is computed as follows:
// Assume `input` has `k` dimensions `[I, J, K, ..., M, N]`, then the output is a // Assume `input` has `k` dimensions `[I, J, K, ..., M, N]`, then the output is a
@ -40144,9 +40142,9 @@ func ResourceApplyMomentumUseNesterov(value bool) ResourceApplyMomentumAttr {
} }
} }
// Update '*var' according to the momentum scheme. Set use_nesterov = True if you // Update '*var' according to the momentum scheme.
// //
// want to use Nesterov momentum. // Set use_nesterov = True if you want to use Nesterov momentum.
// //
// accum = accum * momentum + grad // accum = accum * momentum + grad
// var -= lr * accum // var -= lr * accum
@ -47099,7 +47097,7 @@ func ResourceApplyAdagradV2UpdateSlots(value bool) ResourceApplyAdagradV2Attr {
// Update '*var' according to the adagrad scheme. // Update '*var' according to the adagrad scheme.
// //
// accum += grad * grad // accum += grad * grad
// var -= lr * grad * (1 / sqrt(accum)) // var -= lr * grad * (1 / (sqrt(accum) + epsilon))
// //
// Arguments: // Arguments:
// var_: Should be from a Variable(). // var_: Should be from a Variable().