Commit 129c4fce authored by Eugene Brevdo's avatar Eugene Brevdo Committed by Amit Patankar
Browse files

Additional colocation options and bugfixes for TensorArray 

* colocate_with is now set properly when a TensorArray is passed through a
  while_loop
* added a new argument, "colocate_with_first_write" (default: True; this is
  the current behavior).  If False, the TensorArray is simply placed on the
  device from the context it's constructed in, and no colocation constraints
  are added.

PiperOrigin-RevId: 157643133
parent bf258560
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