TPURoundRobin

publiczna klasa końcowa TPURoundRobin

Równoważenie obciążenia okrężnie na rdzeniach TPU.

Opcja równoważenia obciążenia, która działa okrężnie między rdzeniami TPU.

Ta opcja działa okrężnie pomiędzy liczbami całkowitymi w [0, NumTPUCoresVisiblePerHost]. Jest to przydatne do łączenia się z operacjami TensorFlow, które przyjmują jako dane wejściowe rdzeń TPU, na którym można wykonywać obliczenia, takie jak „TPUPartitionedCall”.

urządzenie_ordinal: Liczba całkowita w [0, NumTPUCoresVisiblePerHost].

Metody publiczne

Wyjście <liczba całkowita>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny TPURoundRobin
utwórz (zakres zakresu )
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPURoundRobin.
Wyjście <liczba całkowita>

Metody dziedziczone

Metody publiczne

publiczne wyjście <liczba całkowita> asOutput ()

Zwraca symboliczny uchwyt tensora.

Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.

public static TPURoundRobin create ( Zakres zakresu)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPURoundRobin.

Parametry
zakres aktualny zakres
Zwroty
  • nowa instancja TPURoundRobin

public Output <Integer> urządzenieOrdinal ()