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