Op thực thi một chương trình với các cập nhật biến tại chỗ tùy chọn.
Nó (tùy chọn) đọc các biến của thiết bị, tải và thực thi chương trình TPU trên thiết bị TPU, sau đó (tùy chọn) cập nhật các biến tại chỗ bằng cách sử dụng đầu ra của chương trình, như được chỉ định trong các thuộc tính device_var_reads_indices (chỉ số đầu vào chương trình từ các biến đọc trực tiếp) và device_var_updates_indices (chỉ số đầu ra chương trình được sử dụng để cập nhật các biến, -1 có nghĩa là không cập nhật/chỉ đọc). Các đầu ra chương trình như vậy được tiêu thụ bởi các biến này sẽ không xuất hiện trong đầu ra op. Để sử dụng nội bộ của trình biên dịch TPU phân tán.
Phương pháp công khai
tĩnh TPUExecuteAndUpdateVariables | |
Trình vòng lặp< Toán hạng <Đối tượng>> | trình lặp () |
Danh sách< Đầu ra <?>> | kết quả () |
Phương pháp kế thừa
Phương pháp công khai
tĩnh công khai TPUExecuteAndUpdateVariables tạo ( Phạm vi phạm vi, Iterable< Toán hạng <?>> args, Phím toán tử <String>, Danh sách<Class<?>> Tresults, Danh sách<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TPUExecuteAndUpdateVariables 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 TPUExecuteAndUpdateVariables