Commit f4f588f4 authored by Gaurav Jain's avatar Gaurav Jain Committed by TensorFlower Gardener
Browse files

Set element_shape for TensorArray in tf.scan 

XLA needs the element_shape to be know to support TensorArray.

PiperOrigin-RevId: 231875046
parent b4150f00
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