Commit 057117af authored by Nick Felt's avatar Nick Felt Committed by TensorFlower Gardener
Browse files

Ensure that tf.summary.write() keeps metadata tensor on CPU 

Previously, when in a GPU device context, the logic attempted to create the default empty metadata tensor on the GPU, which fails since it's a string tensor.

PiperOrigin-RevId: 235135723
parent c80a1ff8
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