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

Remove unnecessary device scope

Browse Source 
PiperOrigin-RevId: 274188668
This commit is contained in:
Gaurav Jain 2019-10-11 09:34:35 -07:00 committed by TensorFlower Gardener
parent ac577de928
commit d1a1b7f4cb
1 changed files with 1 additions and 2 deletions

3
tensorflow/python/ops/array_ops.py
View File

@ -2531,8 +2531,7 @@ def zeros_like_impl(tensor, dtype, name, optimize=True):
if dtype is not None and dtype != tensor.dtype:
return zeros(
shape_internal(tensor, optimize=optimize), dtype=dtype, name=name)
with ops.device(tensor.device):
return gen_array_ops.zeros_like(tensor, name=name)
return gen_array_ops.zeros_like(tensor, name=name)
# For now, variant types must be created via zeros_like; as we need to
# pass the input variant object to the proper zeros callback.