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> | |
Wyjście <T> | wyjście () |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
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