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
wartość logiczna | równa się (Obiekt arg0) |
ostatnia klasa<?> | pobierzKlasę () |
wew | hashCode () |
ostateczna pustka | powiadomić () |
ostateczna pustka | powiadom wszystkich () |
Strunowy | doString () |
ostateczna pustka | czekaj (długi arg0, int arg1) |
ostateczna pustka | czekaj (długi arg0) |
ostateczna pustka | Czekać () |
abstrakcyjne wyjście <String> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
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