Commit da18bea8 authored by Saurabh Saxena's avatar Saurabh Saxena Committed by TensorFlower Gardener
Browse files

Lazily initialize TensorLists passed as input to While if needed. This is... 

Lazily initialize TensorLists passed as input to While if needed. This is similar to the existing setup for TensorArrays.
If an uninitialized list is passed to the XlaWhile, it first compiles the body with the uninitialized list input to get the output list shape. Then it creates an initialized TL using the inferred shape and re-compiles the body.
This enables tf.map_fn in XLA with v2 control flow.

PiperOrigin-RevId: 236871595
parent a75ecd09
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