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 | |
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