InfeedEnqueuePrelinearizedBuffer

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
utwórz (zakres zakresu , wejście argumentu <?>, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację 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.