lớp tĩnh công khai InfeedEnqueueTuple.Options
Thuộc tính tùy chọn cho InfeedEnqueueTuple
Phương pháp công khai
InfeedEnqueueTuple.Options | deviceOrdinal (Thiết bị dàiOrdinal) |
InfeedEnqueueTuple.Options | bố cục (Bố cục danh sách<Long>) |
Phương pháp kế thừa
Phương pháp công khai
public InfeedEnqueueTuple.Options deviceOrdinal (Thiết bị dàiOrdinal)
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. |
---|
bố cục InfeedEnqueueTuple.Options công khai (bố cục Danh sách<Long>)
Thông số
bố cục | Một vectơ giữ bố cục được yêu cầu theo trình tự từ nhỏ đến lớn cho tất cả các hình dạng bộ dữ liệu, theo thứ tự các hình dạng xuất hiện trong đầu vào "hình dạng". Các thành phần bố cục cho hình dạng phụ có thể được đặt thành -1, trong trường hợp đó, bố cục tương ứng sẽ được tính toán bằng thao tác nạp liệu. |
---|