flux tensoriel : : TensorBuffer

Il s'agit d'une classe abstraite.

#include <tensor.h>

Résumé

Interface pour accéder au tampon de données brutes comptées par référence.

Héritage

Hérite de : RefCounted

Constructeurs et Destructeurs

TensorBuffer (void *data_ptr)
~TensorBuffer ()

Fonctions publiques

FillAllocationDescription (AllocationDescription *proto) const =0
virtual void
Remplit les métadonnées sur l'allocation dans le proto.
GetAllocatedBytes (size_t *out_bytes) const
virtual bool
OwnsMemory () const
virtual bool
Indique si ce TensorBuffer possède la mémoire sous-jacente.
base () const
T *
Méthode d'assistance pour réinterpréter le tampon comme un tableau de T .
data () const
void *
data() pointe vers une région mémoire de size() octets.
root_buffer ()=0
virtual TensorBuffer *
Si ce TensorBuffer est un sous-tampon d'un autre TensorBuffer , renvoie ce TensorBuffer .
size () const =0
virtual size_t
Taille (en octets) du tampon.

Fonctions publiques

FillAllocationDescription

virtual void FillAllocationDescription(
  AllocationDescription *proto
) const =0

Remplit les métadonnées sur l'allocation dans le proto.

GetAllocatedBytes

virtual bool GetAllocatedBytes(
  size_t *out_bytes
) const 

PossèdeMémoire

virtual bool OwnsMemory() const 

Indique si ce TensorBuffer possède la mémoire sous-jacente.

TensorBuffer

 TensorBuffer(
  void *data_ptr
)

base

T * base() const 

Méthode d'assistance pour réinterpréter le tampon comme un tableau de T .

données

void * data() const 

data() pointe vers une région mémoire de size() octets.

REMARQUE (mrry) : La méthode data() n'est pas virtuelle pour des raisons de performances. Il peut être appelé plusieurs fois lors de l'accès au contenu d'un Tensor , et ainsi le rendre non virtuel permet au corps d'être intégré.

root_buffer

virtual TensorBuffer * root_buffer()=0

Si ce TensorBuffer est un sous-tampon d'un autre TensorBuffer , renvoie ce TensorBuffer .

Sinon, renvoie ceci.

taille

virtual size_t size() const =0

Taille (en octets) du tampon.

~TensorBuffer

 ~TensorBuffer() override