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

Don't recompute "as_dtype" in backprop._ones

Browse Source 
PiperOrigin-RevId: 238321327
This commit is contained in:
Akshay Modi 2019-03-13 15:22:50 -07:00 committed by TensorFlower Gardener
parent cdb7884729
commit ec35b04906
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/python/eager/backprop.py
View File

@ -624,13 +624,14 @@ def _zeros(shape, dtype):
def _ones(shape, dtype):
if dtypes.as_dtype(dtype) == dtypes.string:
as_dtype = dtypes.as_dtype(dtype)
if as_dtype == dtypes.string:
return None
if not context.context().executing_eagerly():
return array_ops.ones(shape, dtype)
if dtypes.as_dtype(dtype).is_bool:
if as_dtype.is_bool:
value = True
else:
value = 1