org.tensorflow.op.xla

Kelas

BroadcastHelper <T memperluas TType > Operator pembantu untuk melakukan siaran ala XLA

Menyiarkan `lhs` dan `rhs` ke peringkat yang sama, dengan menambahkan dimensi ukuran 1 ke `lhs` dan `rhs` mana pun yang memiliki peringkat lebih rendah, menggunakan aturan penyiaran XLA untuk operator biner.

ClusterOutput <T memperluas TType > Operator yang menghubungkan output komputasi XLA ke node grafik konsumen lainnya.
Konv <T memperluas TType > Membungkus operator XLA ConvGeneralDilated, didokumentasikan di

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

Dekuantisasi Mengambil masukan uint32 yang dikemas dan membongkar masukan tersebut ke uint8 untuk dilakukan

Dekuantisasi pada perangkat.

Titik <T memperluas TType > Membungkus operator XLA DotGeneral, didokumentasikan di

https://www.tensorflow.org/kinerja/xla/operation_semantics#dotgeneral .

DynamicSlice <T memperluas TType > Membungkus operator XLA DynamicSlice, didokumentasikan di

https://www.tensorflow.org/kinerja/xla/operation_semantics#dynamicslice .

DynamicUpdateSlice <T memperluas TType > Membungkus operator XLA DynamicUpdateSlice, didokumentasikan di

https://www.tensorflow.org/kinerja/xla/operation_semantics#dynamicupdateslice .

Einsum <T memperluas TType > Sebuah operasi yang mendukung operasi einsum dasar dengan 2 input dan 1 output.
Kumpulkan <T extends TType > Membungkus operator XLA Gather yang didokumentasikan di

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

KeyValueSort <T memperluas TNumber , U memperluas TType > Membungkus operator Sortir XLA, didokumentasikan di

https://www.tensorflow.org/kinerja/xla/operation_semantics#sort .

Pad <T memperluas TType > Membungkus operator XLA Pad, didokumentasikan di

https://www.tensorflow.org/kinerja/xla/operation_semantics#pad .

Penerimaan <T memperluas TType > Menerima tensor bernama dari komputasi XLA lain.
ReplikaId ID replika.
SelfAdjointEig <T memperluas TType > Menghitung dekomposisi eigen dari sekumpulan matriks self-adjoint

(Catatan: Hanya masukan nyata yang didukung).

Mengirim Mengirimkan tensor bernama ke perhitungan XLA lainnya.
Berbagi <T memperluas TType > Sebuah operasi yang membagi input berdasarkan atribut sharding yang diberikan.
Urutkan <T extends TType > Membungkus operator Sortir XLA, didokumentasikan di

https://www.tensorflow.org/kinerja/xla/operation_semantics#sort .

Svd <T memperluas TType > Menghitung dekomposisi eigen dari sekumpulan matriks self-adjoint

(Catatan: Hanya masukan nyata yang didukung).

XlaRecvFromHost <T memperluas TType > Sebuah operasi untuk menerima tensor dari host.
XlaSendToHost Sebuah operasi untuk mengirim tensor ke host.
XlaSetBound Tetapkan batasan untuk nilai masukan yang diberikan sebagai petunjuk untuk kompiler Xla,

mengembalikan nilai yang sama.