tensorflow::TensorBuffer

This is an abstract class.

#include <tensor.h>

Summary

Interface to access the raw ref-counted data buffer.

Inheritance

Inherits from: RefCounted

Constructors and Destructors
TensorBuffer(void *data_ptr)
~TensorBuffer()

Public functions
FillAllocationDescription(AllocationDescription *proto) const =0
virtual void
Fills metadata about the allocation into the proto.
GetAllocatedBytes(size_t *out_bytes) const
virtual bool
GetMemoryType() const
virtual AllocatorMemoryType
The type of the underlying memory.
OwnsMemory() const
virtual bool
Whether this TensorBuffer owns the underlying memory.
base() const
T *
Helper method to reinterpret the buffer as an array of T.
data() const
void *
data() points to a memory region of size() bytes.
root_buffer()=0
virtual TensorBuffer *
If this TensorBuffer is sub-buffer of another TensorBuffer, returns that TensorBuffer.
size() const =0
virtual size_t
Size (in bytes) of the buffer.

Public functions

FillAllocationDescription

virtual void FillAllocationDescription(
  AllocationDescription *proto
) const =0

Fills metadata about the allocation into the proto.

GetAllocatedBytes

virtual bool GetAllocatedBytes(
  size_t *out_bytes
) const

GetMemoryType

virtual AllocatorMemoryType GetMemoryType() const

The type of the underlying memory.

OwnsMemory

virtual bool OwnsMemory() const

Whether this TensorBuffer owns the underlying memory.

TensorBuffer

 TensorBuffer(
  void *data_ptr
)

base

T * base() const

Helper method to reinterpret the buffer as an array of T.

data

void * data() const

data() points to a memory region of size() bytes.

NOTE(mrry): The data() method is not virtual for performance reasons. It can be called multiple times when the contents of a Tensor are accessed, and so making it non-virtual allows the body to be inlined.

root_buffer

virtual TensorBuffer * root_buffer()=0

If this TensorBuffer is sub-buffer of another TensorBuffer, returns that TensorBuffer.

Otherwise, returns this.

size

virtual size_t size() const =0

Size (in bytes) of the buffer.

~TensorBuffer

 ~TensorBuffer() override