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

שיטות בירושה

מכיתה org.tensorflow.op.PrimitiveOp

בוליאנית סופית	שווה (Object obj)
int סופי	hashcode ()
מבצע	אופ () מחזירה את Operation הבסיסית
מחרוזת סופית	toString ()

מכיתה java.lang.Object

בוליאני	שווה (Object arg0)
שיעור אחרון<?>	getClass ()
int	hashcode ()
ריק סופי	להודיע ​​()
ריק סופי	הודע הכל ()
חוּט	toString ()
ריק סופי	המתן (ארג0 ארוך, int arg1)
ריק סופי	חכה (ארג0 ארוך)
ריק סופי	חכה ()

מהממשק java.lang.Iterable

בָּטֵל	forEach (Consumer<? super T> arg0)
איטרטור מופשט< Oprand <Object>>	איטרטור ()
Spliterator< Operand <Object>>	מפצל ()

שיטות ציבוריות

ציבורי סטטי MlirPassthroughOp create (היקף היקף , כניסות Iterable< Operand <?>>, String mlirModule, List<Class<?>> Toutputs)

שיטת מפעל ליצירת מחלקה העוטפת פעולת MlirPassthroughOp חדשה.

פרמטרים

תְחוּם	ההיקף הנוכחי

החזרות

• מופע חדש של MlirPassthroughOp

איטרטור ציבורי< Operand <Object>> איטרטור ()

רשימה ציבורית< פלט <?>> פלטים ()