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

Remove usage of internal ops.numpy_text by forking it into Keras.

Browse Source 
PiperOrigin-RevId: 339102563
Change-Id: Ic21e266529dcd44e2a2199bbd0391beef4444a79
This commit is contained in:
Tomer Kaftan 2020-10-26 13:12:13 -07:00 committed by TensorFlower Gardener
parent b5359acda5
commit 76a08f796c
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tensorflow/python/keras/mixed_precision/autocast_variable.py
View File

@ -34,6 +34,19 @@ from tensorflow.python.types import core
_autocast_dtype = threading.local()
def numpy_text(tensor, is_repr=False):
"""Human readable representation of a tensor's numpy value."""
if tensor.dtype.is_numpy_compatible:
# pylint: disable=protected-access
text = repr(tensor._numpy()) if is_repr else str(tensor._numpy())
# pylint: enable=protected-access
else:
text = '<unprintable>'
if '\n' in text:
text = '\n' + text
return text
class AutoCastVariable(variables.Variable, core.Tensor):
"""Variable that will cast itself to a different dtype in applicable contexts.
@ -144,7 +157,7 @@ class AutoCastVariable(variables.Variable, core.Tensor):
'dtype={v.dtype.name} dtype_to_cast_to={v._cast_dtype.name}, '
'numpy={np_repr}>')
return repr_str.format(
v=self, np_repr=ops.numpy_text(self.read_value(), is_repr=True))
v=self, np_repr=numpy_text(self.read_value(), is_repr=True))
else:
repr_str = ("<AutoCastVariable '{v.name}' shape={v.shape} "
'dtype={v.dtype.name} dtype_to_cast_to={v._cast_dtype.name}>')
@ -534,5 +547,3 @@ class enable_auto_cast_variables(object): # pylint:disable=invalid-name
def __exit__(self, type_arg, value_arg, traceback_arg):
_autocast_dtype.dtype = self._prev_dtype