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

Reverse change eb75ded6 so that internal tests will pass. (#12933)

Browse Source 
As support for int64 global steps is not ready in TPUs, I am reversing this change so that our internal performance and regression tests will pass.
This commit is contained in:
Frank Chen 2017-09-09 10:03:19 -07:00 committed by Yifei Feng
parent 32ffc5a81e
commit 877c9deca3
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/contrib/tpu/python/tpu/tpu_estimator.py
View File

@ -68,11 +68,12 @@ def _create_global_step(graph):
return variable_scope.get_variable( return variable_scope.get_variable(
ops.GraphKeys.GLOBAL_STEP, ops.GraphKeys.GLOBAL_STEP,
shape=[], shape=[],
dtype=dtypes.int64, dtype=dtypes.int32,
initializer=init_ops.zeros_initializer(), initializer=init_ops.zeros_initializer(),
trainable=False, trainable=False,
use_resource=True, use_resource=True,
collections=[ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.GLOBAL_STEP]) collections=[ops.GraphKeys.GLOBAL_VARIABLES,
ops.GraphKeys.GLOBAL_STEP])
def _sync_variables_ops(): def _sync_variables_ops():