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. |