Commit 2e707494 authored by Guangda Lai's avatar Guangda Lai Committed by TensorFlower Gardener
Browse files

Fix BaseGPUDevice, let it report the actual memory limit of the allocator.... 

Fix BaseGPUDevice, let it report the actual memory limit of the allocator. Also added a helper method to reset ProcessState.

PiperOrigin-RevId: 186655996
parent 51b33487
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