NcclAllReduce

publiczna klasa końcowa NcclAllReduce

Wysyła tensor zawierający redukcję we wszystkich tensorach wejściowych.

Wysyła tensor zawierający redukcję wszystkich tensorów wejściowych przekazanych do operacji w ramach tej samej „nazwy_wspólnej”.

Wykres powinien zostać skonstruowany w taki sposób, że jeśli jedna operacja zostanie uruchomiona z wartością nazwa_wspólna o wartości „c”, wówczas liczba operacji „liczba_urządzeń” zostanie uruchomiona z nazwą nazwa_wspólna o wartości „c”. Niezastosowanie się do tego spowoduje, że wykonanie wykresu nie zostanie ukończone.

wejście: wejście do danych redukcji: wartość redukcji na wszystkich urządzeniach `num_devices`. redukcja: operacja redukcji do wykonania. num_devices: Liczba urządzeń objętych tą obniżką. wspólna_nazwa: Identyfikator wspólny dla operacji o tej samej redukcji.

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny <T numer rozszerzenia> NcclAllReduce <T>
utwórz (zakres zakresu , wejście argumentu <T>, redukcja ciągu, długa liczba urządzeń, ciąg wspólna nazwa)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację NcclAllReduce.
Wyjście <T>
dane ()

Metody dziedziczone

Metody publiczne

publiczne wyjście <T> 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 NcclAllReduce <T> create ( zakres zakresu , argument wejściowy <T>, redukcja ciągu, długa liczba urządzeń, ciąg wspólna nazwa)

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

Parametry
zakres aktualny zakres
Zwroty
  • nowa instancja NcclAllReduce

publiczne dane wyjściowe <T> ()