experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update tf.MatrixBandPart, tf.ResourceApplyAdagradV2 and tf.ResourceApplyMomentum summaries and descriptions to match between TensorFlow op definitions and TensorFlow MLIR ODS.

Browse Source 
PiperOrigin-RevId: 308338605
Change-Id: I31c240b984ac4f1f873379910682d935ad50ac07
This commit is contained in:
Andy Ly 2020-04-24 15:29:50 -07:00 committed by TensorFlower Gardener
parent f5d2eabea0
commit 64b83e8724
4 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
View File

@ -3979,8 +3979,7 @@ cublas.
def TF_MatrixBandPartOp : TF_Op<"MatrixBandPart", [AllTypesMatch<["input", "band"]>, NoSideEffect]> {
let summary = [{
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.
}];
let description = [{

4
tensorflow/core/api_def/base_api/api_def_MatrixBandPart.pbtxt
View File

@ -26,10 +26,8 @@ END
Rank `k` tensor of the same shape as input. The extracted banded tensor.
END
}
summary: "Copy a tensor setting everything outside a central band in each innermost matrix"
summary: "Copy a tensor setting everything outside a central band in each innermost matrix to zero."
description: <<END
to zero.
The `band` part is computed as follows:
Assume `input` has `k` dimensions `[I, J, K, ..., M, N]`, then the output is a
tensor with the same shape where

2
tensorflow/core/api_def/base_api/api_def_ResourceApplyAdagradV2.pbtxt
View File

@ -42,6 +42,6 @@ END
summary: "Update \'*var\' according to the adagrad scheme."
description: <<END
accum += grad * grad
var -= lr * grad * (1 / sqrt(accum))
var -= lr * grad * (1 / (sqrt(accum) + epsilon))
END
}

4
tensorflow/core/api_def/base_api/api_def_ResourceApplyMomentum.pbtxt
View File

@ -46,9 +46,9 @@ var - lr * momentum * accum, so in the end, the var you get is actually
var - lr * momentum * accum.
END
}
summary: "Update \'*var\' according to the momentum scheme. Set use_nesterov = True if you"
summary: "Update \'*var\' according to the momentum scheme."
description: <<END
want to use Nesterov momentum.
Set use_nesterov = True if you want to use Nesterov momentum.
accum = accum * momentum + grad
var -= lr * accum