lớp cuối cùng công khai EnqueueTPUEmbeddingIntegerBatch
Một op sắp xếp danh sách các tensor lô đầu vào vào TPUEmbedding.
Các lớp lồng nhau
lớp học | EnqueueTPUEmbeddingIntegerBatch.Options | Thuộc tính tùy chọn cho EnqueueTPUEmbeddingIntegerBatch |
Phương pháp công khai
EnqueueTPUEembeddingIntegerBatch tĩnh | tạo ( Phạm vi phạm vi, lô Iterable< Operand <Integer>>, Operand <String> modeOverride, 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 EnqueueTPUEmbeddingIntegerBatch mới. |
EnqueueTPUEmbeddingIntegerBatch.Options tĩnh | deviceOrdinal (Thiết bị dàiOrdinal) |
Phương pháp kế thừa
Phương pháp công khai
public static EnqueueTPUEmbeddingIntegerBatch create ( Phạm vi phạm vi, Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Options... options)
Phương thức gốc để tạo một lớp bao bọc một thao tác EnqueueTPUEmbeddingIntegerBatch mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
lô hàng | Danh sách các tensor 1D, một cho mỗi bảng nhúng, chứa các chỉ số trong các bảng. |
chế độghi đè | Đầu vào chuỗi ghi đè chế độ được chỉ định trong TPUEmbeddingConfiguration. Các giá trị được hỗ trợ là {'unspecified', 'inference', 'training', 'backward_pass_only'}. Khi được đặt thành 'không xác định', chế độ được đặt trong TPUEmbeddingConfiguration sẽ được sử dụng, nếu không thì mode_override sẽ được sử dụng. |
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 EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
Thông số
thiết bịthứ tự | Thiết bị TPU sẽ sử dụng. Phải >= 0 và nhỏ hơn số lõi TPU trong tác vụ mà nút được đặt trên đó. |
---|