Classi
BroadcastHelper <T estende TType > | Operatore di supporto per l'esecuzione di trasmissioni in stile XLA Trasmette "lhs" e "rhs" allo stesso rango, aggiungendo dimensioni di dimensione 1 a quello tra "lhs" e "rhs" che ha il rango inferiore, utilizzando le regole di trasmissione XLA per gli operatori binari. |
ClusterOutput <T estende TType > | Operatore che collega l'output di un calcolo XLA ad altri nodi del grafico consumer. |
Conv <T estende TType > | Avvolge l'operatore XLA ConvGeneralDilated, documentato in https://www.tensorflow.org/performance/xla/Operation_semantics#conv_convolution. |
Dequantizzare | Prende l'input uint32 compresso e decomprime l'input in uint8 per farlo Dequantizzazione sul dispositivo. |
Punto <T estende TType > | Avvolge l'operatore XLA DotGeneral, documentato in https://www.tensorflow.org/performance/xla/operazione_semantics#dotgeneral . |
DynamicSlice <T estende TType > | Avvolge l'operatore XLA DynamicSlice, documentato in https://www.tensorflow.org/performance/xla/Operation_semantics#dynamicslice. |
DynamicUpdateSlice <T estende TType > | Avvolge l'operatore XLA DynamicUpdateSlice, documentato in https://www.tensorflow.org/performance/xla/operazione_semantics#dynamicupdateslice. |
Einsum <T estende TType > | Un'operazione che supporta l'operazione einsum di base con 2 ingressi e 1 uscita. |
Raccogli <T estende TType > | Avvolge l'operatore XLA Gather documentato in https://www.tensorflow.org/xla/Operation_semantics#gather |
KeyValueSort <T estende TNumber , U estende TType > | Avvolge l'operatore XLA Sort, documentato in https://www.tensorflow.org/performance/xla/operazione_semantica#sort. |
Pad <T estende TType > | Avvolge l'operatore XLA Pad, documentato a https://www.tensorflow.org/performance/xla/operazione_semantica#pad. |
Recv <T estende TType > | Riceve il tensore denominato da un altro calcolo XLA. |
ID replica | Identificazione della replica. |
SelfAdjointEig <T estende TType > | Calcola la scomposizione automatica di un lotto di matrici autoaggiunte (Nota: sono supportati solo input reali). |
Inviare | Invia il tensore denominato a un altro calcolo XLA. |
Sharding <T estende TType > | Un'operazione che suddivide l'input in base all'attributo di sharding specificato. |
Ordina <T estende TType > | Avvolge l'operatore XLA Sort, documentato in https://www.tensorflow.org/performance/xla/operazione_semantica#sort. |
Svd <T estende TType > | Calcola la scomposizione automatica di un lotto di matrici autoaggiunte (Nota: sono supportati solo input reali). |
XlaRecvFromHost <T estende TType > | Un'operazione per ricevere un tensore dall'host. |
XlaSendToHost | Un'operazione per inviare un tensore all'host. |
XlaSetBound | Imposta un limite per il valore di input fornito come suggerimento per il compilatore Xla, restituisce lo stesso valore. |