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

org.tensorflow.op.PrimitiveOp
końcowa wartość logiczna
równa się (obiekt obiektu)
końcowy wew
Operacja
op ()
Zwraca podstawową Operation
ostatni ciąg
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ć ()
org.tensorflow.Operand
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