Commit a102a6a7 authored by Gaurav Jain's avatar Gaurav Jain Committed by TensorFlower Gardener
Browse files

Add use_gpu, force_gpu & force_gpu helpers in test_util 

In order to make tests compatible for V2 behavior we should avoid using
sessions and instead use device contexts. To support the different
desired behaviors we add 3 helpers in test_util and covert a few tests
to make use of them.

PiperOrigin-RevId: 222333755
parent 96b0d298
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