מחלקה סופית ציבורית InfeedEnqueuePrelinearizedBuffer
אופציה שמעמידה בתור מאגר מאופיין בהזנת TPU.
כיתות מקוננות
מעמד | InfeedEnqueuePrelinearizedBuffer.Options | תכונות אופציונליות עבור InfeedEnqueuePrelinearizedBuffer |
שיטות ציבוריות
static InfeedEnqueuePrelinearizedBuffer | ליצור ( היקף היקף , קלט Operand <?>, אפשרויות... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת InfeedEnqueuePrelinearizedBuffer חדשה. |
static InfeedEnqueuePrelinearizedBuffer.Options | deviceOrdinal (Long deviceOrdinal) |
שיטות בירושה
שיטות ציבוריות
public static InfeedEnqueuePrelinearizedBuffer create ( היקף היקף , קלט Operand <?>, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת InfeedEnqueuePrelinearizedBuffer חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
קֶלֶט | טנזור משתנה המייצג פלט ליניארי. |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של InfeedEnqueuePrelinearizedBuffer
public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)
פרמטרים
deviceOrdinal | מכשיר ה-TPU לשימוש. זה צריך להיות -1 כאשר ה-Op פועל על התקן TPU ו-=0 כאשר ה-Op פועל על התקן ה-CPU. |
---|