텐서플로우:: 텐서버퍼

이것은 추상 클래스입니다.

#include <tensor.h>

요약

원시 참조 계산 데이터 버퍼에 액세스하기 위한 인터페이스입니다.

계승

상속 대상: RefCounted

생성자와 소멸자

TensorBuffer (void *data_ptr)
~TensorBuffer ()

공공 기능

FillAllocationDescription (AllocationDescription *proto) const =0
virtual void
proto에 할당에 대한 메타데이터를 채웁니다.
GetAllocatedBytes (size_t *out_bytes) const
virtual bool
OwnsMemory () const
virtual bool
TensorBuffer가 기본 메모리를 소유하는지 여부입니다.
base () const
T *
버퍼를 T 배열로 재해석하는 도우미 메서드입니다.
data () const
void *
data()는 size() 바이트의 메모리 영역을 가리킵니다.
root_buffer ()=0
virtual TensorBuffer *
TensorBuffer가 다른 TensorBuffer 의 하위 버퍼인 경우 해당 TensorBuffer를 반환합니다.
size () const =0
virtual size_t
버퍼의 크기(바이트)입니다.

공공 기능

채우기할당설명

virtual void FillAllocationDescription(
  AllocationDescription *proto
) const =0

proto에 할당에 대한 메타데이터를 채웁니다.

할당된바이트 가져오기

virtual bool GetAllocatedBytes(
  size_t *out_bytes
) const 

메모리 소유

virtual bool OwnsMemory() const 

TensorBuffer가 기본 메모리를 소유하는지 여부입니다.

텐서버퍼

 TensorBuffer(
  void *data_ptr
)

베이스

T * base() const 

버퍼를 T 배열로 재해석하는 도우미 메서드입니다.

데이터

void * data() const 

data()는 size() 바이트의 메모리 영역을 가리킵니다.

참고(mrry): data() 메서드는 성능상의 이유로 가상이 아닙니다. Tensor 의 콘텐츠에 액세스할 때 여러 번 호출할 수 있으므로 이를 비가상으로 만들면 본문이 인라인될 수 있습니다.

루트_버퍼

virtual TensorBuffer * root_buffer()=0

TensorBuffer가 다른 TensorBuffer 의 하위 버퍼인 경우 해당 TensorBuffer를 반환합니다.

그렇지 않으면 이것을 반환합니다.

크기

virtual size_t size() const =0

버퍼의 크기(바이트)입니다.

~텐서버퍼

 ~TensorBuffer() override