AllReduce

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

Giảm lẫn nhau nhiều tensor có cùng loại và hình dạng.

Các lớp lồng nhau

lớp học AllReduce.Options Thuộc tính tùy chọn cho AllReduce

Hằng số

Sợi dây OP_NAME Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Phương pháp công khai

Đầu ra <T>
asOutput ()
Trả về tay cầm tượng trưng của tensor.
tĩnh AllReduce.Options
CommunicationsHint (Giao tiếp chuỗiHint)
tĩnh <T mở rộng TNumber > AllReduce <T>
tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Kích thước nhóm dài, Khóa nhóm dài, Khóa trường hợp dài, Chuỗi hợp nhấtOp, Chuỗi cuối cùng, Danh sách <Long> subdivOffsets, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác AllReduce mới.
Đầu ra <T>
tĩnh AllReduce.Options
hết thời gian chờGiây (Thời gian chờ nổiGiây)
tĩnh AllReduce.Options
chờFor (Danh sách <Dài> chờ)

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

Hằng số

Chuỗi cuối cùng tĩnh công khai OP_NAME

Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Giá trị không đổi: "CollectiveReduce"

Phương pháp công khai

Đầu ra công khai <T> asOutput ()

Trả về tay cầm tượng trưng của tensor.

Đầ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.

public static AllReduce.Options CommunicationsHint (String CommunicationsHint)

public static AllReduce <T> tạo ( Phạm vi phạm vi , đầu vào Toán hạng <T>, Long groupSize, Long groupKey, Long instanceKey, String mergeOp, String FinalOp, List<Long> subdivOffsets, Options... tùy chọn)

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

Thông số
phạm vi phạm vi hiện tại
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 AllReduce

Dữ liệu đầu ra công khai <T> ()

public static AllReduce.Options timeoutSeconds (Thời gian chờ nổiGiây)

tĩnh công khai AllReduce.Options waitFor (Danh sách<Long> waitFor)