Commit 587808a8 authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

test_util.py: Allow use_gpu to change between calls to self.cached_session() 

use_gpu does not affect the creation of the session, it only affects the
context manager in which nodes are added to the graph, so it should not
be included in the consistency check.

PiperOrigin-RevId: 211659833
parent 5032036e
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