XlaSpmdShardToFullShape

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

Một op được sử dụng bởi trình phân vùng XLA SPMD để chuyển từ phân vùng thủ công sang

phân vùng tự động. Nó chuyển đổi đầu vào được phân vùng thủ công, có hình dạng phân đoạn thành tensor có hình dạng đầy đủ để được phân vùng tự động với cùng một phân đoạn được sử dụng bằng cách phân vùng thủ công.

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 <T mở rộng TType > XlaSpmdShardToFullShape <T>
tạo (Phạm vi phạm vi , đầu vào Toán hạng <T>, Chuỗi thủ côngSharding, Hình dạng fullShape)
Phương thức ban đầu để tạo một lớp bao gồm thao tác XlaSpmdShardToFullShape mới.
Đầu ra <T>
đầu ra ()

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: "XlaSpmdShardToFullShape"

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 XlaSpmdShardToFullShape <T> tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Chuỗi thủ côngSharding, Hình dạng fullShape)

Phương thức ban đầu để tạo một lớp bao gồm thao tác XlaSpmdShardToFullShape 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 XlaSpmdShardToFullShape

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