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

Merge pull request #39717 from SwapnilDreams100:master 

PiperOrigin-RevId: 312847670
Change-Id: I7f2d4a5cf63b55ce6d013ceb49a30ab342efa7bf
This commit is contained in:
TensorFlower Gardener 2020-05-22 12:20:01 -07:00
parent 591330b967 18a7a327f3
commit 19f4ac7694
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/python/distribute/tpu_strategy.py
View File

@ -897,7 +897,7 @@ class TPUExtended(distribute_lib.StrategyExtendedV1):
if tensor_util.is_tensor(input_tensor):
rank = input_tensor.get_shape().rank
else:
rank = np.rank(input_tensor)
rank = np.ndim(input_tensor)
maximum_shape = tensor_shape.TensorShape([None] * rank)
maximum_shapes.append(maximum_shape)
maximum_shapes = nest.pack_sequence_as(replicate_inputs[0],