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)
}
// Copy a tensor setting everything outside a central band in each innermost matrix
//
// to zero.
// Copy a tensor setting everything outside a central band in each innermost matrix to zero.
//
// The `band` part is computed as follows:
// 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
// var -= lr * accum
@ -47099,7 +47097,7 @@ func ResourceApplyAdagradV2UpdateSlots(value bool) ResourceApplyAdagradV2Attr {
// Update '*var' according to the adagrad scheme.
//
// accum += grad * grad
// var -= lr * grad * (1 / sqrt(accum))
// var -= lr * grad * (1 / (sqrt(accum) + epsilon))
//
// Arguments:
// var_: Should be from a Variable().