org.tensorflow.op.xla

Zajęcia

BroadcastHelper <T rozszerza TType > Operator pomocniczy do wykonywania transmisji w stylu XLA

Rozgłasza `lhs` i `rhs` do tej samej rangi, dodając wymiary o rozmiarze 1 do tego, który z `lhs` i `rhs` ma niższą rangę, używając reguł rozgłaszania XLA dla operatorów binarnych.

ClusterOutput <T rozszerza TType > Operator łączący wynik obliczenia XLA z innymi węzłami wykresu konsumenckiego.
Konw <T rozszerza TType > Zawija operator XLA ConvGeneralDilated, udokumentowany pod adresem

https://www.tensorflow.org/ Performance/xla/operative_semantics#conv_convolution .

Dekwantyzacja Pobiera spakowane dane wejściowe uint32 i rozpakowuje je do uint8 w celu wykonania

Dekwantyzacja na urządzeniu.

Kropka <T rozszerza TType > Zawija operator XLA DotGeneral, udokumentowany pod adresem

https://www.tensorflow.org/ Performance/xla/operative_semantics#dotgeneral .

DynamicSlice <T rozszerza TType > Zawija operator XLA DynamicSlice, udokumentowany pod adresem

https://www.tensorflow.org/ Performance/xla/operative_semantics#dynamicslice .

DynamicUpdateSlice <T rozszerza TType > Zawija operator XLA DynamicUpdateSlice, udokumentowany pod adresem

https://www.tensorflow.org/ Performance/xla/operative_semantics#dynamicupdateslice .

Einsum <T rozszerza TType > Operacja obsługująca podstawową operację einsum z 2 wejściami i 1 wyjściem.
Zbierz <T rozszerza TType > Zawija operator XLA Gather udokumentowany pod adresem

https://www.tensorflow.org/xla/operative_semantics#gather

KeyValueSort <T rozszerza TNumber , U rozszerza TType > Zawija operator sortowania XLA, udokumentowany pod adresem

https://www.tensorflow.org/ Performance/xla/operative_semantics#sort .

Pad <T rozszerza TType > Zawija operatora XLA Pad, udokumentowany pod adresem

https://www.tensorflow.org/ Performance/xla/operative_semantics#pad .

Odbiór <T rozszerza TType > Otrzymuje nazwany tensor z innego obliczenia XLA.
Identyfikator repliki Identyfikator repliki.
SelfAdjointEig <T rozszerza TType > Oblicza rozkład własny partii macierzy samosprzężonych

(Uwaga: obsługiwane są tylko wejścia rzeczywiste).

Wysłać Wysyła nazwany tensor do innego obliczenia XLA.
Fragmentowanie <T rozszerza TType > Operacja, która dzieli dane wejściowe na podstawie danego atrybutu fragmentowania.
Sortuj <T rozszerza TType > Zawija operator sortowania XLA, udokumentowany pod adresem

https://www.tensorflow.org/ Performance/xla/operative_semantics#sort .

Svd <T rozszerza TType > Oblicza rozkład własny partii macierzy samosprzężonych

(Uwaga: obsługiwane są tylko wejścia rzeczywiste).

XlaRecvFromHost <T rozszerza TType > Opcja otrzymania tensora od hosta.
XlaSendToHost Opcja wysłania tensora do hosta.
XlaSetBound Ustaw granicę dla podanej wartości wejściowej jako wskazówkę dla kompilatora Xla,

zwraca tę samą wartość.