Home
last modified time | relevance | path

Searched refs:GetAllocatedBytes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dwindow_dataset.cc47 allocated_bytes += GetAllocatedBytes(element); in AllocatedBytes()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h568 int64 GetAllocatedBytes(const std::vector<Tensor>& element);
761 node_->add_buffered_bytes(-GetAllocatedBytes(element)); in RecordBufferDequeue()
770 node_->add_buffered_bytes(GetAllocatedBytes(element)); in RecordBufferEnqueue()
Ddataset.cc319 int64 GetAllocatedBytes(const std::vector<Tensor>& element) { in GetAllocatedBytes() function