From ec35b04906ba08e2252065103ff638a33b21ac6d Mon Sep 17 00:00:00 2001 From: Akshay Modi Date: Wed, 13 Mar 2019 15:22:50 -0700 Subject: [PATCH] Don't recompute "as_dtype" in backprop._ones PiperOrigin-RevId: 238321327 --- tensorflow/python/eager/backprop.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/eager/backprop.py b/tensorflow/python/eager/backprop.py index 801e3d6fa56..0bce60d6997 100644 --- a/tensorflow/python/eager/backprop.py +++ b/tensorflow/python/eager/backprop.py @@ -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