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

[XLA:Python] Fix Numpy deprecation warning for use of np.object.

Browse Source 
Will fix https://github.com/google/jax/issues/4424 when included in a jaxlib.

PiperOrigin-RevId: 334611917
Change-Id: I9dc76f812d4744567b62d4b5cf385ded375c4986
This commit is contained in:
Peter Hawkins 2020-09-30 09:10:26 -07:00 committed by TensorFlower Gardener
parent fa75523767
commit f61e1203bd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/python/xla_client.py
View File

@ -193,8 +193,8 @@ XLA_ELEMENT_TYPE_TO_DTYPE = {
PrimitiveType.F64: np.dtype('float64'), PrimitiveType.F64: np.dtype('float64'),
PrimitiveType.C64: np.dtype('complex64'), PrimitiveType.C64: np.dtype('complex64'),
PrimitiveType.C128: np.dtype('complex128'), PrimitiveType.C128: np.dtype('complex128'),
PrimitiveType.TUPLE: np.dtype(np.object), PrimitiveType.TUPLE: np.dtype(np.object_),
PrimitiveType.TOKEN: np.dtype(np.object), PrimitiveType.TOKEN: np.dtype(np.object_),
} }
# Note the conversion on the key. Numpy has a known issue wherein dtype hashing # Note the conversion on the key. Numpy has a known issue wherein dtype hashing