tensorflow:: TensorBuffer

Il s'agit d'une classe abstraite.

#include <tensor.h>

Résumé

Interface permettant d'accéder à la mémoire tampon de données brutes à comptage de références.

Héritage

Hérite de : RefCounted

Constructeurs et destructeurs

TensorBuffer (void *data_ptr)
~TensorBuffer ()

fonctions publiques

FillAllocationDescription (AllocationDescription *proto) const =0
virtual void
Renseigne les métadonnées relatives à l'allocation dans le proto.
GetAllocatedBytes (size_t *out_bytes) const
virtual bool
OwnsMemory () const
virtual bool
Indique si ce TensorBuffer est propriétaire de la mémoire sous-jacente.
base () const
T *
Méthode auxiliaire pour réinterpréter le tampon comme un tableau de T
data () const
void *
data() pointe vers une région mémoire de taille() 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

Description de l'allocation de remplissage

virtual void FillAllocationDescription(
  AllocationDescription *proto
) const =0

Renseigne les métadonnées relatives à l'allocation dans le proto.

Obtenir les octets alloués

virtual bool GetAllocatedBytes(
  size_t *out_bytes
) const 

Possède la mémoire

virtual bool OwnsMemory() const 

Indique si ce TensorBuffer est propriétaire de la mémoire sous-jacente.

TensorBuffer

 TensorBuffer(
  void *data_ptr
)

base

T * base() const 

Méthode auxiliaire 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 taille() octets.

NOTE (mrry) : La méthode data() n’est pas virtuelle pour des raisons de performance. Elle peut être appelée plusieurs fois lors de l’accès au contenu d’un Tensor ; la rendre non virtuelle permet donc d’intégrer son corps directement dans le code.

tampon_racine

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