עוטף חישוב 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>
'''
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
סטטי MlirPassthroughOp | |
Iterator< Operand < TType >> | איטרטור () |
רשימה< פלט <?>> | פלטים () |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
ציבורי סטטי MlirPassthroughOp create (היקף היקף , כניסות Iterable< Operand <?>>, String mlirModule, List<Class<? מרחיב את TType >> Toutputs)
שיטת מפעל ליצירת מחלקה העוטפת פעולת MlirPassthroughOp חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|
מחזיר
- מופע חדש של MlirPassthroughOp