Fix BaseGPUDevice, let it report the actual memory limit of the allocator. Also added...
authorGuangda Lai <laigd@google.com>
Thu, 22 Feb 2018 20:24:22 +0000 (12:24 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 22 Feb 2018 20:28:36 +0000 (12:28 -0800)
commit2e707494c4b1058e1186c67b1030f635bdf52dac
tree41c878ed54dcd311c83527a460a4d2b8f79af990
parent51b334875125b1e76545d02d2e8e18c7ff2be0af
Fix BaseGPUDevice, let it report the actual memory limit of the allocator. Also added a helper method to reset ProcessState.

PiperOrigin-RevId: 186655996
tensorflow/core/common_runtime/gpu/gpu_device.cc