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

Adding uint32 to the the supported infeed dtypes

Browse Source 
PiperOrigin-RevId: 231617395
This commit is contained in:
Daryl Ng 2019-01-30 10:14:58 -08:00 committed by TensorFlower Gardener
parent 3599af7b15
commit 609992a3a0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/tpu/python/ops/tpu_ops.py
View File

@ -157,7 +157,7 @@ if platform.system() != "Windows":
_SUPPORTED_INFEED_DTYPES = set([ _SUPPORTED_INFEED_DTYPES = set([
dtypes.bool, dtypes.int32, dtypes.int64, dtypes.bfloat16, dtypes.float32, dtypes.bool, dtypes.int32, dtypes.int64, dtypes.bfloat16, dtypes.float32,
dtypes.complex64 dtypes.complex64, dtypes.uint32
]) ])
def infeed_dequeue(dtype, shape, name=None): def infeed_dequeue(dtype, shape, name=None):