OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fast-path for EagerTensorBase.dtype

Browse Source 
PiperOrigin-RevId: 170933005
This commit is contained in:
Alexandre Passos 2017-10-03 16:24:33 -07:00 committed by TensorFlower Gardener
parent 08e266d9b5
commit b925f8553c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/framework/ops.py
View File

@ -582,7 +582,9 @@ class _EagerTensorBase(Tensor):
@property
def dtype(self):
return dtypes.as_dtype(self._datatype_enum())
# Note: using the intern table directly here as this is
# performance-sensitive in some models.
return dtypes._INTERN_TABLE[self._datatype_enum()] # pylint: disable=protected-access
def _numpy_text(self, is_repr=False):
if self.dtype.is_numpy_compatible: