Operacja wykonująca partycję TPUEmbedding w konfiguracji centralnej
urządzenia i oblicza rozmiar HBM (w bajtach) wymagany do operacji TPUEmbedding.
Metody publiczne
Wyjście <Ciąg> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
Wyjście <Ciąg> | wspólna konfiguracja () Wspólny proto konfiguracji zakodowany w postaci ciągu, zawierający metadane dotyczące wyjścia partycjonatora TPUEmbedding i rozmiaru HBM (w bajtach) wymaganego do działania. |
statyczny WykonajTPUEmbeddingPartitioner |
Metody dziedziczone
Metody publiczne
publiczne wyjście <String> 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.
publiczne wyjście <String> commonConfig ()
Wspólny proto konfiguracji zakodowany w postaci ciągu, zawierający metadane dotyczące wyjścia partycjonatora TPUEmbedding i rozmiaru HBM (w bajtach) wymaganego do działania.
public static ExecuteTPUEmbeddingPartitioner create (zakres zakresu , konfiguracja ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ExecuteTPUEmbeddingPartitioner.
Parametry
zakres | aktualny zakres |
---|---|
konfiguracja | Proto TPUEmbeddingConfiguration serializowane do ciągu znaków opisującego pożądaną konfigurację TPUEmbedding. |
Zwroty
- nowa instancja ExecuteTPUEmbeddingPartitioner