Commit b0c21b65 authored by Eugene Brevdo's avatar Eugene Brevdo Committed by TensorFlower Gardener
Browse files

TensorArray now uses colocation instead of lazy device setters. Added test helper. 

* Move a new test method making it easy to write local distributed unit tests
  into the tf.test namespace.
* Update the rnn helper functions to use colocation in a few places as well.
* Use it to test the new colocation behavior.

PiperOrigin-RevId: 156254308
parent d726686d
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