publiczna klasa końcowa ExecuteTPUEmbeddingPartitioner
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