Commit a76646d4 authored by Yunxing Dai's avatar Yunxing Dai Committed by TensorFlower Gardener
Browse files

Add type checking at the beginning of tpu.shard(). 

Otherwise a message like "TypeError: Tensor objects are only iterable when eager execution is enabled. To iterate over this tensor use tf.map_fn." will be thrown, which is confusing.

PiperOrigin-RevId: 213371676
parent 185aa899
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment