publiczna klasa końcowa InfeedEnqueuePrelinearizedBuffer
Operacja kolejkująca wstępnie zlinearyzowany bufor do zasilania TPU.
Klasy zagnieżdżone
klasa | InfeedEnqueuePrelinearizedBuffer.Options | Opcjonalne atrybuty dla InfeedEnqueuePrelinearizedBuffer |
Metody publiczne
statyczny InfeedEnqueuePrelinearizedBuffer | |
statyczny InfeedEnqueuePrelinearizedBuffer.Options | urządzenieOrdinal (Długie urządzenieOrdinal) |
Metody dziedziczone
Metody publiczne
public static InfeedEnqueuePrelinearizedBuffer create (zakres zakresu , wejście argumentu <?>, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację InfeedEnqueuePrelinearizedBuffer.
Parametry
zakres | aktualny zakres |
---|---|
wejście | Wariant tensora reprezentujący zlinearyzowane wyjście. |
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja InfeedEnqueuePrelinearizedBuffer
public static InfeedEnqueuePrelinearizedBuffer.Options urządzenieOrdinal (Długie urządzenieOrdinal)
Parametry
urządzeniePorządek | Urządzenie TPU, którego chcesz użyć. Powinno to wynosić -1, gdy Op działa na urządzeniu TPU i = 0, gdy Op działa na urządzeniu CPU. |
---|