org.tensorflow.op.xla

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.