ConfigureTPUEmbeddingMemory

publiczna klasa końcowa ConfigureTPUEmbeddingMemory

Operacja konfigurująca oprogramowanie TPUEmbedding na hoście.

Metody publiczne

Wyjście <Ciąg>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny SkonfigurujTPUEmbeddingMemory
utwórz ( Zakres zasięgu, Operand <String> commonConfig)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ConfigureTPUEmbeddingMemory.
Wyjście <Ciąg>
Konfiguracja pamięci ()
Konfiguracja pamięci zakodowana w postaci ciągu zawierająca metadane dotyczące alokacji pamięci zarezerwowanych dla TPUEmbedding.

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.

public static ConfigureTPUEmbeddingMemory create ( Zakres zakresu, Operand <String> commonConfig)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ConfigureTPUEmbeddingMemory.

Parametry
zakres aktualny zakres
wspólna konfiguracja Proto CommonConfiguration zakodowane w postaci ciągu zawierające metadane dotyczące wyjścia partycjonatora TPUEmbedding i rozmiaru HBM (w bajtach) wymaganego do działania.
Zwroty
  • nowa instancja ConfigureTPUEmbeddingMemory

publiczne wyjście <String> memoryConfig ()

Konfiguracja pamięci zakodowana w postaci ciągu zawierająca metadane dotyczące alokacji pamięci zarezerwowanych dla TPUEmbedding.