Commit 3cd2d2a6 authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

Make CriticalSection work inside a Dataset with eager execution enabled. 

tf.colocate_with() might be provided with eager tensors when
constructing TensorFlow functions (like the subgraph for map()
inside a tf.data.Dataset).

Prior to this change, the added test would fail with:
"Tensor.op is meaningless when eager execution is enabled."

PiperOrigin-RevId: 194217166
parent ffd34990
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