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