MlirPassthroughOp

MlirPassthroughOp คลาสสุดท้ายสาธารณะ

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

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

{@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>
 }
 '''

วิธีการสาธารณะ

MlirPassthroughOp แบบคงที่
สร้าง (ขอบเขต ขอบเขต , อินพุต < ตัวดำเนินการ <?>> ที่สามารถทำซ้ำได้, สตริง mlirModule, รายการ <Class<?>> Toutputs)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ MlirPassthroughOp ใหม่
ตัววนซ้ำ < ตัวดำเนินการ <วัตถุ>>
รายการ < เอาท์พุต <?>>

วิธีการสืบทอด

วิธีการสาธารณะ

การสร้าง MlirPassthroughOp แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต < ตัวดำเนินการ <?>>, สตริง mlirModule, รายการ <Class<?>> Toutputs)

วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ MlirPassthroughOp ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
การส่งคืน
  • อินสแตนซ์ใหม่ของ MlirPassthroughOp

ตัววนซ้ำสาธารณะ < ตัวดำเนินการ <วัตถุ>> ตัววนซ้ำ ()

รายการสาธารณะ < เอาท์พุต <?>> เอาท์พุต ()