MlirPassthroughOp

ล้อมการคำนวณ MLIR ตามอำเภอใจที่แสดงเป็นโมดูลด้วยฟังก์ชัน main()

การดำเนินการนี้ไม่มีเคอร์เนลที่เกี่ยวข้องและไม่ได้ตั้งใจให้ดำเนินการในเซสชัน TensorFlow ปกติ แต่มีวัตถุประสงค์เพื่อใช้สำหรับการทดสอบหรือสำหรับกรณีพิเศษที่ผู้ใช้ตั้งใจที่จะส่งผ่านการคำนวณ MLIR แบบกำหนดเองผ่านกราฟ TensorFlow โดยมีจุดประสงค์เพื่อให้เครื่องมือแบบกำหนดเองประมวลผลดาวน์สตรีม (เมื่อกำหนดเป้าหมายสภาพแวดล้อมอื่น เช่น TensorFlow lite เป็นต้น) โมดูล MLIR คาดว่าจะมีฟังก์ชัน main() ที่จะใช้เป็นจุดเริ่มต้น อินพุตสำหรับการดำเนินการจะถูกส่งผ่านเป็นอาร์กิวเมนต์ไปยังฟังก์ชัน main() และค่าที่ส่งคืนของฟังก์ชันหลักที่แมปกับเอาต์พุต ตัวอย่างการใช้งาน:

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>
 
 '''