diff options
| author |  Mark Heffernan <meheff@google.com> | 2017-05-03 10:13:00 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-03 11:24:45 -0700 |
| commit | 427081c11b55d494b92fa836f67ad07439096115 (patch) | |
| tree | 72f838df8695a2063f839c7f6fc2be13f097da35 /tools | |
| parent | 44446379790a211df6c77cf7dff8ad819cd4ef57 (diff) | |
[XLA] Account for computation output size correctly in HLO rematerialization.
Previously in HLO rematerialization, the tuple elements of the output were not counted against memory use. This change fixes this. Also, move the accounting for the parameter to the caller of HLO rematerialization because the arguments to the computation are existing allocation prior to execution and are not part of the memory allocated specifically for the computation.
Change: 154978404
Diffstat (limited to 'tools')
0 files changed, 0 insertions, 0 deletions