InfeedEnqueuePrelinearizedBuffer

classe finale publique InfeedEnqueuePrelinearizedBuffer

Une opération qui met en file d'attente le tampon prélinéarisé dans l'alimentation TPU.

Classes imbriquées

classe InfeedEnqueuePrelinearizedBuffer.Options Attributs facultatifs pour InfeedEnqueuePrelinearizedBuffer

Méthodes publiques

statique InfeedEnqueuePrelinearizedBuffer
créer (portée de portée , entrée opérande <?>, options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération InfeedEnqueuePrelinearizedBuffer.
statique InfeedEnqueuePrelinearizedBuffer.Options
appareilOrdinal (appareilOrdinal long)

Méthodes héritées

Méthodes publiques

public static InfeedEnqueuePrelinearizedBuffer créer (portée de portée , entrée opérande <?>, options... options)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération InfeedEnqueuePrelinearizedBuffer.

Paramètres
portée portée actuelle
saisir Un tenseur variant représentant une sortie linéarisée.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de InfeedEnqueuePrelinearizedBuffer

public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)

Paramètres
appareilOrdinal Le périphérique TPU à utiliser. Cela doit être -1 lorsque l'opération s'exécute sur un périphérique TPU et = 0 lorsque l'opération s'exécute sur le périphérique CPU.