ExecuteTPUEmbeddingPartitioner

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
utwórz (zakres zakresu , konfiguracja ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ExecuteTPUEmbeddingPartitioner.

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