diff options
author | 2018-01-25 14:53:37 -0800 | |
---|---|---|
committer | 2018-01-25 14:57:31 -0800 | |
commit | ad5c04c9e1151c4de71288520d45f3b3142299fb (patch) | |
tree | cab5d270404d8b9a3a84bb0c7e6d2dd563d8cc0b | |
parent | b998b7b456066530dd27ef532dae195d27505266 (diff) |
Whitelist "bool" as a valid TPU infeed type.
PiperOrigin-RevId: 183296017
-rw-r--r-- | tensorflow/contrib/tpu/python/ops/tpu_ops.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tpu/python/ops/tpu_ops.py b/tensorflow/contrib/tpu/python/ops/tpu_ops.py index a49a3dcf29..1c970655d0 100644 --- a/tensorflow/contrib/tpu/python/ops/tpu_ops.py +++ b/tensorflow/contrib/tpu/python/ops/tpu_ops.py @@ -47,7 +47,7 @@ if platform.system() != "Windows": # types are supported. _SUPPORTED_INFEED_DTYPES = set([ - dtypes.int32, dtypes.bfloat16, dtypes.float32 + dtypes.bool, dtypes.int32, dtypes.bfloat16, dtypes.float32 ]) def infeed_dequeue(dtype, shape, name=None): |