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ụ:
{@code 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> } '''
Phương pháp công khai
Mlir tĩnhTruyền quaOp | |
Trình vòng lặp< Toán hạng <Đối tượng>> | trình lặp () |
Danh sách< Đầu ra <?>> | đầu ra () |
Phương pháp kế thừa
Phương pháp công khai
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<?>> 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