TPURoundRobin

lớp cuối cùng công khai TPURoundRobin

Cân bằng tải theo vòng tròn trên lõi TPU.

Một cơ chế cân bằng tải xoay vòng giữa các lõi TPU.

Hoạt động này sẽ quay vòng giữa các số nguyên trong [0, NumTPUCoresVisiblePerHost]. Nó rất hữu ích khi giao tiếp với các hoạt động TensorFlow lấy lõi TPU làm đầu vào để thực hiện các phép tính, chẳng hạn như `TPUPartitionedCall`.

device_orderal: Một số nguyên trong [0, NumTPUCoresVisiblePerHost].

Phương pháp công khai

Đầu ra <Số nguyên>
asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
TPURoundRobin tĩnh
tạo ( Phạm vi phạm vi)
Phương thức xuất xưởng để tạo một lớp bao bọc hoạt động TPURoundRobin mới.
Đầu ra <Số nguyên>
thiết bịOrdinal ()

Phương pháp kế thừa

Phương pháp công khai

Đầu ra công khai <Số nguyên> asOutput ()

Trả về phần điều khiển tượng trưng của một tenxơ.

Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.

tạo TPURoundRobin tĩnh công khai ( Phạm vi phạm vi)

Phương thức xuất xưởng để tạo một lớp bao bọc hoạt động TPURoundRobin mới.

Thông số
phạm vi phạm vi hiện tại
Trả lại
  • một phiên bản mới của TPURoundRobin

Đầu ra công khai <Integer> deviceOrdinal ()