Des classes
| BroadcastHelper <T étend TType > | Opérateur assistant pour effectuer des diffusions de style XLA Diffuse `lhs` et `rhs` au même rang, en ajoutant des dimensions de taille 1 à celui de `lhs` et `rhs` ayant le rang inférieur, en utilisant les règles de diffusion de XLA pour les opérateurs binaires. |
| ClusterOutput <T étend TType > | Opérateur qui connecte la sortie d'un calcul XLA à d'autres nœuds de graphe consommateur. |
| Conv <T étend TType > | Encapsule l'opérateur XLA ConvGeneralDilated, documenté dans https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution . |
| Déquantifier | Prend l'entrée uint32 compressée et décompresse l'entrée dans uint8 pour le faire Déquantification sur l'appareil. |
| Point <T étend TType > | Encapsule l'opérateur XLA DotGeneral, documenté dans https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral . |
| DynamicSlice <T étend TType > | Encapsule l'opérateur XLA DynamicSlice, documenté sur https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice . |
| DynamicUpdateSlice <T étend TType > | Encapsule l'opérateur XLA DynamicUpdateSlice, documenté sur https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice . |
| Einsum <T étend TType > | Une opération qui prend en charge une opération de base avec 2 entrées et 1 sortie. |
| Rassembler <T étend TType > | Encapsule l'opérateur XLA Gather documenté dans https://www.tensorflow.org/xla/operation_semantics#gather |
| KeyValueSort <T étend TNumber , U étend TType > | Encapsule l'opérateur XLA Sort, documenté dans https://www.tensorflow.org/performance/xla/operation_semantics#sort . |
| Pad <T étend TType > | Enveloppe l'opérateur XLA Pad, documenté sur https://www.tensorflow.org/performance/xla/operation_semantics#pad . |
| Recv <T étend TType > | Reçoit le tenseur nommé d'un autre calcul XLA. |
| ID de réplica | ID de réplique. |
| SelfAdjointEig <T étend TType > | Calcule la décomposition propre d'un lot de matrices auto-adjointes (Remarque : seules les entrées réelles sont prises en charge). |
| Envoyer | Envoie le tenseur nommé à un autre calcul XLA. |
| Sharding <T étend TType > | Une opération qui fragmente l'entrée en fonction de l'attribut de partitionnement donné. |
| Trier <T étend TType > | Encapsule l'opérateur XLA Sort, documenté dans https://www.tensorflow.org/performance/xla/operation_semantics#sort . |
| Svd <T étend TType > | Calcule la décomposition propre d'un lot de matrices auto-adjointes (Remarque : seules les entrées réelles sont prises en charge). |
| XlaRecvFromHost <T étend TType > | Une opération pour recevoir un tenseur de l'hôte. |
| XlaSendToHost | Une opération pour envoyer un tenseur à l'hôte. |
| XlaSetBound | Définissez une limite pour la valeur d'entrée donnée comme indice pour le compilateur Xla, renvoie la même valeur. |