diff options
| author |  Yuefeng Zhou <yuefengz@google.com> | 2017-03-02 14:36:17 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-02 14:48:36 -0800 |
| commit | fee34c128479c5b1549ae2635b3f034bda526441 (patch) | |
| tree | dc394cce6104f0b27d81b67c77c8fd17bdc3655f /tensorflow/core/framework/op_kernel.h | |
| parent | 2f9c1d2d205e1b7be111dd87a26d7c3a4d57c6c1 (diff) | |
Tracking memory usage of queues.
Change: 149048750
Diffstat (limited to 'tensorflow/core/framework/op_kernel.h')
| -rw-r--r-- | tensorflow/core/framework/op_kernel.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/framework/op_kernel.h b/tensorflow/core/framework/op_kernel.h index 5d0a2e3d24..1d5a46a5e6 100644 --- a/tensorflow/core/framework/op_kernel.h +++ b/tensorflow/core/framework/op_kernel.h @@ -207,6 +207,8 @@ class PersistentTensor { int64 NumElements() const { return tensor_.NumElements(); } + int64 TotalBytes() const { return tensor_.TotalBytes(); } + private: Tensor tensor_; }; |