Một op được sử dụng bởi trình phân vùng XLA SPMD để chuyển từ phân vùng tự động sang
phân vùng thủ công. Nó chú thích đầu vào (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ởi phân vùng thủ công và xuất ra một tenxơ hình phân đoạn để sử dụng cho các hoạt động được phân vùng thủ công sau này. Nếu hình dạng không thể phân chia đồng đều, vùng đệm sẽ được che bằng 0.
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 > XlaSpmdFullToShardShape <T> | |
Đầ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
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 XlaSpmdFullToShardShape <T> tạo (Phạm vi phạm vi , đầu vào Toán tử <T>, Chuỗi thủ côngSharding)
Phương thức ban đầu để tạo một lớp bao gồm thao tác XlaSpmdFullToShardShape 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 XlaSpmdFullToShardShape