NcclAllReduce

Clase public final NcclAllReduce

Genera un tensor que contiene la reducción en todos los tensores de entrada.

Genera un tensor que contiene la reducción en todos los tensores de entrada pasados ​​a operaciones dentro del mismo `nombre_compartido.

El gráfico debe construirse de modo que si una operación se ejecuta con el valor de nombre_compartido `c`, entonces las operaciones de` num_devices` se ejecutarán con el valor de nombre_compartido `c`. Si no lo hace, la ejecución del gráfico no se completará.

input: la entrada a los datos de reducción: el valor de la reducción en todos los dispositivos `num_devices`. reducción: la operación de reducción a realizar. num_devices: la cantidad de dispositivos que participan en esta reducción. shared_name: Identificador que se comparte entre operaciones de la misma reducción.

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

Salida <T>
asOutput ()
Devuelve el identificador simbólico del tensor.
estática <T se extiende TNumber > NcclAllReduce <T>
crear ( Alcance alcance, operando <T> de entrada, la reducción String, numDevices largo, cadena sharedName)
Método de fábrica para crear una clase que envuelva una nueva operación NcclAllReduce.
Salida <T>
datos ()

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "NcclAllReduce"

Métodos públicos

pública de salida <T> asOutput ()

Devuelve el identificador simbólico del tensor.

Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.

public static NcclAllReduce <T> crear ( Alcance alcance, operando <T> de entrada, la reducción de cuerdas, numDevices largas, Cadena sharedName)

Método de fábrica para crear una clase que envuelva una nueva operación NcclAllReduce.

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de NcclAllReduce

pública de salida <T> de datos ()