org.tensorflow.op.xla

Các lớp học

BroadcastHelper <T mở rộng TType > Toán tử trợ giúp để thực hiện các chương trình phát sóng kiểu XLA

Phát sóng `lhs` và `rhs` đến cùng một thứ hạng, bằng cách thêm kích thước 1 vào bất kỳ thứ nguyên nào trong số `lhs` và `rhs` có thứ hạng thấp hơn, sử dụng quy tắc phát sóng của XLA cho toán tử nhị phân.

ClusterOutput <T mở rộng TType > Toán tử kết nối đầu ra của phép tính XLA với các nút biểu đồ người tiêu dùng khác.
Chuyển đổi <T mở rộng TType > Bao bọc toán tử XLA ConvGeneralDilating, được ghi lại tại

https://www.tensorflow.org/performance/xla/Operation_semantics#conv_convolution.

giải mã Lấy đầu vào uint32 đã đóng gói và giải nén đầu vào thành uint8 để thực hiện

Dequantization trên thiết bị.

Dấu chấm <T mở rộng TType > Bao bọc toán tử XLA DotGeneral, được ghi lại tại

https://www.tensorflow.org/performance/xla/Operation_semantics#dotgeneral .

DynamicSlice <T mở rộng TType > Bao bọc toán tử XLA DynamicSlice, được ghi lại tại

https://www.tensorflow.org/performance/xla/Operation_semantics#dynamicslice.

DynamicUpdateSlice <T mở rộng TType > Bao bọc toán tử XLA DynamicUpdateSlice, được ghi lại tại

https://www.tensorflow.org/performance/xla/Operation_semantics#dynamicupdateslice .

Einsum <T mở rộng TType > Một op hỗ trợ op einsum cơ bản với 2 đầu vào và 1 đầu ra.
Tập hợp <T mở rộng TType > Bao bọc toán tử XLA Gather được ghi lại tại

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

KeyValueSort <T mở rộng TNumber , U mở rộng TType > Bao bọc toán tử Sắp xếp XLA, được ghi lại tại

https://www.tensorflow.org/performance/xla/Operation_semantics#sort.

Pad <T mở rộng TType > Bao bọc toán tử XLA Pad, được ghi lại tại

https://www.tensorflow.org/performance/xla/Operation_semantics#pad .

Recv <T mở rộng TType > Nhận tenxơ được đặt tên từ một phép tính XLA khác.
Id bản sao ID bản sao.
SelfAdjointEig <T mở rộng TType > Tính toán phân rã riêng của một loạt ma trận tự liên kết

(Lưu ý: Chỉ hỗ trợ đầu vào thực).

Gửi Gửi tenxơ được đặt tên tới một tính toán XLA khác.
Phân mảnh <T mở rộng TType > Một op phân chia đầu vào dựa trên thuộc tính phân chia đã cho.
Sắp xếp <T mở rộng TType > Bao bọc toán tử Sắp xếp XLA, được ghi lại tại

https://www.tensorflow.org/performance/xla/Operation_semantics#sort.

Svd <T mở rộng TType > Tính toán phân rã riêng của một loạt ma trận tự liên kết

(Lưu ý: Chỉ hỗ trợ đầu vào thực).

XlaRecvFromHost <T mở rộng TType > Một lệnh để nhận tensor từ máy chủ.
XlaGửi tới máy chủ Một lệnh gửi một tensor tới máy chủ.
XlaSetBound Đặt giới hạn cho giá trị đầu vào đã cho làm gợi ý cho trình biên dịch Xla,

trả về cùng một giá trị.