Bao bọc một phép tính MLIR tùy ý được biểu thị dưới dạng một mô-đun có hàm main().
Hoạt động này không có hạt nhân liên quan và không nhằm mục đích thực thi trong phiên TensorFlow thông thường. Thay vào đó, nó được thiết kế để sử dụng để thử nghiệm hoặc cho trường hợp đặc biệt khi người dùng dự định chuyển tính toán MLIR tùy chỉnh thông qua biểu đồ TensorFlow với mục đích yêu cầu công cụ tùy chỉnh xử lý nó ở phía dưới (khi nhắm mục tiêu một môi trường khác, chẳng hạn như TensorFlow lite). Mô-đun MLIR dự kiến sẽ có hàm main() sẽ được sử dụng làm điểm vào. Đầu vào của các thao tác sẽ được chuyển dưới dạng đối số cho hàm main() và các giá trị trả về của hàm chính được ánh xạ tới đầu ra. Cách sử dụng ví dụ:
import tensorflow as tf
from tensorflow.compiler.mlir.tensorflow.gen_mlir_passthrough_op import mlir_passthrough_op
mlir_module = '''python
func @main(%arg0 : tensor<10xf32>, %arg1 : tensor<10xf32>) -> tensor<10x10xf32> {
%add = "magic.op"(%arg0, %arg1) : (tensor<10xf32>, tensor<10xf32>) -> tensor<10x10xf32>
return %ret : tensor<10x10xf32>
'''
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 cộng
Mlir tĩnhTruyền quaOp | |
Trình vòng lặp< Toán hạng < TType >> | trình lặp () |
Danh sách< Đầu ra <?>> | đầ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 cộng
công khai tĩnh MlirPassthroughOp tạo ( Phạm vi phạm vi, Iterable< Toán hạng <?>> đầu vào, Chuỗi mlirModule, Danh sách<Class<? mở rộng TType >> Toutputs)
Phương thức gốc để tạo một lớp bao bọc một thao tác MlirPassthroughOp 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 MlirPassthroughOp