diff options
| author |  Anna R <annarev@google.com> | 2018-09-12 12:29:19 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-12 12:38:24 -0700 |
| commit | 28e945e590b07de137f318a70896bc4fc31f7053 (patch) | |
| tree | 59435a08e14420e284ac1c0aed60074bd2a71435 /tensorflow/stream_executor/stream_executor_pimpl.h | |
| parent | f337425dc71e3ea95aa91ce401a40c1b594486ca (diff) | |
Internal change.
PiperOrigin-RevId: 212684548
Diffstat (limited to 'tensorflow/stream_executor/stream_executor_pimpl.h')
| -rw-r--r-- | tensorflow/stream_executor/stream_executor_pimpl.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/stream_executor_pimpl.h b/tensorflow/stream_executor/stream_executor_pimpl.h index 437f298616..d04025b681 100644 --- a/tensorflow/stream_executor/stream_executor_pimpl.h +++ b/tensorflow/stream_executor/stream_executor_pimpl.h @@ -699,6 +699,13 @@ class StreamExecutor { // The set of TraceListeners registered for this StreamExecutor. std::set<TraceListener*> listeners_ GUARDED_BY(mu_); + // Allocated memory in bytes. + int64 mem_alloc_bytes_; + + // Memory limit in bytes. Value less or equal to 0 indicates there is no + // limit. + int64 memory_limit_bytes_; + SE_DISALLOW_COPY_AND_ASSIGN(StreamExecutor); }; |