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

Update tpu_embedding.py

Browse Source
This commit is contained in:
geetachavan1 2021-03-03 11:36:24 -08:00 committed by GitHub
parent 20b3e15b74
commit e87c70aef4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/tpu/tpu_embedding.py
View File

@ -329,8 +329,8 @@ ProximalAdagradSlotVariableName = collections.namedtuple(
FtrlSlotVariableName = collections.namedtuple('FtrlSlotVariableName',
['accumulator', 'linear'])
ProximalYogiSlotVariableNames = collections.namedtuple('ProximalYogiSlotVariableNames',
['v', 'm'])
ProximalYogiSlotVariableNames = collections.namedtuple(
'ProximalYogiSlotVariableNames', ['v', 'm'])
AdamSlotVariables = collections.namedtuple(
'AdamSlotVariables', ['m', 'v'])