Conozca lo último en aprendizaje automático, IA generativa y más en el
Simposio WiML 2023.
NcclAllReduce
Organiza tus páginas con colecciones
Guarda y categoriza el contenido según tus preferencias.
Produce un tensor que contiene la reducción en todos los tensores de entrada.
Muestra un tensor que contiene la reducción en todos los tensores de entrada pasados a operaciones dentro del mismo `shared_name.
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: El número de dispositivos que participan en esta reducción. shared_name: Identificador que se comparte entre operaciones de la misma reducción.
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
estático <T extiende Número> NcclAllReduce <T> | crear ( ámbito de ámbito, entrada de operando <T>, reducción de cadena, número largo de dispositivos, nombre compartido de cadena) Método de fábrica para crear una clase que envuelve una nueva operación NcclAllReduce. |
Salida <T> | |
Métodos Heredados
De la clase java.lang.Object booleano | es igual a (Objeto arg0) |
clase final<?> | obtenerClase () |
En t | código hash () |
vacío final | notificar () |
vacío final | notificar a todos () |
Cuerda | a la cadena () |
vacío final | espera (largo arg0, int arg1) |
vacío final | esperar (largo arg0) |
vacío final | espera () |
Métodos públicos
salida pública <T> como salida ()
Devuelve el identificador simbólico de un tensor.
Las entradas de 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.
Método de fábrica para crear una clase que envuelve una nueva operación NcclAllReduce.
Devoluciones
- una nueva instancia de NcclAllReduce
Salvo que se indique lo contrario, el contenido de esta página está sujeto a la licencia Atribución 4.0 de Creative Commons, y los ejemplos de código están sujetos a la licencia Apache 2.0. Para obtener más información, consulta las políticas del sitio de Google Developers. Java es una marca registrada de Oracle o sus afiliados. Cierto contenido está sujeto a la licencia de NumPy.
Última actualización: 2023-03-31 (UTC)
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Falta la información que necesito"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Muy complicado o demasiados pasos"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Desactualizado"
},{
"type": "thumb-down",
"id": "translationIssue",
"label":"Problema de traducción"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Problema con las muestras o los códigos"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Otro"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Fácil de comprender"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Resolvió mi problema"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Otro"
}]