Commit 9126444b authored by Yuefeng Zhou's avatar Yuefeng Zhou Committed by TensorFlower Gardener
Browse files

Fix memory tracking in the case where temp memory is used as output memory. 

Track all persistent memory allocation in allocate_persistent call except for variables and queues where persistent memory is tracked in variables ops and queue ops. Deallocation of persistent memory is ignored.

PiperOrigin-RevId: 184768231
parent 87b5c8f0
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