tensorflow::TensorBuffer

Đây là một lớp trừu tượng.

#include <tensor.h>

Bản tóm tắt

Giao diện để truy cập vào bộ đệm dữ liệu thô được đếm tham chiếu.

Sự thừa kế

Kế thừa từ: RefCounted

Hàm khởi tạo và hàm hủy

TensorBuffer (void *data_ptr)
~TensorBuffer ()

Chức năng công cộng

FillAllocationDescription (AllocationDescription *proto) const =0
virtual void
Điền siêu dữ liệu về việc phân bổ vào bản ghi proto.
GetAllocatedBytes (size_t *out_bytes) const
virtual bool
OwnsMemory () const
virtual bool
Liệu TensorBuffer này có sở hữu bộ nhớ bên dưới hay không.
base () const
T *
Phương thức hỗ trợ để diễn giải lại bộ đệm dưới dạng một mảng T
data () const
void *
data() trỏ đến một vùng nhớ có kích thước size() byte.
root_buffer ()=0
virtual TensorBuffer *
Nếu TensorBuffer này là bộ đệm con của một TensorBuffer khác, nó sẽ trả về TensorBuffer đó.
size () const =0
virtual size_t
Kích thước (tính bằng byte) của bộ đệm.

Chức năng công cộng

Mô tả phân bổ

virtual void FillAllocationDescription(
  AllocationDescription *proto
) const =0

Điền siêu dữ liệu về việc phân bổ vào bản ghi proto.

Lấy các byte đã phân bổ

virtual bool GetAllocatedBytes(
  size_t *out_bytes
) const 

Bộ nhớ sở hữu

virtual bool OwnsMemory() const 

Liệu TensorBuffer này có sở hữu bộ nhớ bên dưới hay không.

TensorBuffer

 TensorBuffer(
  void *data_ptr
)

căn cứ

T * base() const 

Phương thức hỗ trợ để diễn giải lại bộ đệm dưới dạng một mảng T

dữ liệu

void * data() const 

data() trỏ đến một vùng nhớ có kích thước size() byte.

LƯU Ý (mrry): Phương thức data() không phải là ảo vì lý do hiệu năng. Nó có thể được gọi nhiều lần khi truy cập nội dung của một Tensor , do đó việc không sử dụng phương thức ảo cho phép nội tuyến phần thân của phương thức.

bộ đệm gốc

virtual TensorBuffer * root_buffer()=0

Nếu TensorBuffer này là bộ đệm con của một TensorBuffer khác, nó sẽ trả về TensorBuffer đó.

Ngược lại, nó sẽ trả về giá trị này.

kích cỡ

virtual size_t size() const =0

Kích thước (tính bằng byte) của bộ đệm.

~TensorBuffer

 ~TensorBuffer() override