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 : RefCountedConstructeurs 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 * |
root_buffer ()=0 | virtual 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