Une opération utilisée par le partitionneur XLA SPMD pour passer du partitionnement automatique à
partitionnement manuel. Il annote l'entrée (forme complète, à partitionner automatiquement) avec le même partitionnement que celui utilisé par le partitionnement manuel, et génère un tenseur en forme de fragment qui sera consommé par les opérations ultérieures partitionnées manuellement. Si la forme n'est pas partageable de manière uniforme, la région de remplissage sera masquée avec des 0.
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
Sortie <T> | comme Sortie () Renvoie le handle symbolique du tenseur. |
statique <T étend TType > XlaSpmdFullToShardShape <T> | |
Sortie <T> | sortir () |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static XlaSpmdFullToShardShape <T> créer ( portée de portée , entrée opérande <T>, chaîne manualSharding)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaSpmdFullToShardShape.
Paramètres
portée | portée actuelle |
---|
Retour
- une nouvelle instance de XlaSpmdFullToShardShape