Restez organisé à l'aide des collections Enregistrez et classez les contenus selon vos préférences.

tensorflow :: TensorBuffer

Ceci est 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 relatives à l'allocation dans le proto.
OwnsMemory () const
virtual bool
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 relatives à l'allocation dans le proto.

PossèdeMémoire

virtual bool OwnsMemory() const 

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 sous la forme d'un tableau de T

Les données

void * data() const 

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

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

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