XlaSpmdFullToShardShape

publiczna klasa końcowa XlaSpmdFullToShardShape

Opcja używana przez partycjoner XLA SPMD do przełączania z automatycznego partycjonowania na

ręczne partycjonowanie. Adnotuje dane wejściowe (pełny kształt, które mają zostać automatycznie podzielone) za pomocą tego samego fragmentowania, co przy ręcznym partycjonowaniu, i wyprowadza tensor w kształcie fragmentu, który będzie używany przez późniejsze operacje partycjonowane ręcznie. Jeśli kształtu nie można równomiernie podzielić na partycje, obszar dopełnienia zostanie zamaskowany zerami.

Stałe

Strunowy OP_NAME Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny <T rozszerza TType > XlaSpmdFullToShardShape <T>
utwórz (zakres zakresu , wejście argumentu <T>, ręczne fragmentowanie ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaSpmdFullToShardShape.
Wyjście <T>

Metody dziedziczone

Stałe

publiczny statyczny końcowy ciąg znaków OP_NAME

Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Wartość stała: „XlaSpmdFullToShardShape”

Metody publiczne

publiczne wyjście <T> 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 XlaSpmdFullToShardShape <T> utwórz ( zakres zakresu , wejście argumentu <T>, ręczne fragmentowanie ciągu)

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

Parametry
zakres aktualny zakres
Zwroty
  • nowa instancja XlaSpmdFullToShardShape

publiczne wyjście <T> wyjście ()