lớp cuối cùng công khai InfeedEnqueuePrelineizedBuffer
Một phương án sắp xếp bộ đệm đã được tuyến tính hóa trước vào tiến trình TPU.
Các lớp lồng nhau
| lớp học | InfeedEnqueuePrelineizedBuffer.Options | Các thuộc tính tùy chọn cho InfeedEnqueuePrelinearizedBuffer | |
Phương pháp công khai
| InfeedEnqueuePrelineizedBuffer tĩnh | tạo (Phạm vi phạm vi , đầu vào Toán hạng <?>, Tùy chọn... tùy chọn) Phương thức gốc để tạo một lớp bao bọc một thao tác InfeedEnqueuePrelineizedBuffer mới. |
| tĩnh InfeedEnqueuePrelineizedBuffer.Options | deviceOrdinal (Thiết bị dàiOrdinal) |
Phương pháp kế thừa
Phương pháp công khai
public static InfeedEnqueuePrelineizedBuffer tạo (Phạm vi phạm vi , đầu vào Toán hạng <?>, Tùy chọn... tùy chọn)
Phương thức gốc để tạo một lớp bao bọc một thao tác InfeedEnqueuePrelineizedBuffer mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| đầu vào | Một tensor biến thể đại diện cho đầu ra được tuyến tính hóa. |
| tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của InfeedEnqueuePrelineizedBuffer
public static InfeedEnqueuePrelineizedBuffer.Options deviceOrdinal (Long deviceOrdinal)
Thông số
| thiết bịthứ tự | Thiết bị TPU sẽ sử dụng. Giá trị này phải là -1 khi Op đang chạy trên thiết bị TPU và = 0 khi Op đang chạy trên thiết bị CPU. |
|---|