Clases
BroadcastHelper <T extiende TType > | Operador auxiliar para realizar transmisiones estilo XLA Transmite `lhs` y `rhs` con el mismo rango, agregando dimensiones de tamaño 1 a cualquiera de los `lhs` y `rhs` que tenga el rango inferior, utilizando las reglas de transmisión de XLA para operadores binarios. |
Salida de clúster <T extiende TType > | Operador que conecta la salida de un cálculo XLA a otros nodos del gráfico de consumo. |
Conv <T extiende TType > | Envuelve el operador XLA ConvGeneralDilated, documentado en https://www.tensorflow.org/rendimiento/xla/operación_semantics#conv_convolution. |
Descuantificar | Toma la entrada uint32 empaquetada y la descomprime en uint8 para realizar Descuantificación en el dispositivo. |
Punto <T extiende TType > | Envuelve el operador XLA DotGeneral, documentado en https://www.tensorflow.org/rendimiento/xla/operación_semantics#dotgeneral. |
DynamicSlice <T extiende TType > | Envuelve el operador XLA DynamicSlice, documentado en https://www.tensorflow.org/rendimiento/xla/operación_semantics#dynamicslice. |
DynamicUpdateSlice <T extiende TType > | Envuelve el operador XLA DynamicUpdateSlice, documentado en https://www.tensorflow.org/rendimiento/xla/operación_semantics#dynamicupdateslice. |
Einsum <T extiende TType > | Una operación que admite operaciones básicas einsum con 2 entradas y 1 salida. |
Reunir <T extiende TType > | Envuelve el operador XLA Gather documentado en https://www.tensorflow.org/xla/ Operation_semantics#gather |
KeyValueSort <T extiende TNumber , U extiende TType > | Envuelve el operador XLA Sort, documentado en https://www.tensorflow.org/rendimiento/xla/operación_semantics#sort. |
Pad <T extiende TType > | Envuelve al operador de la almohadilla XLA, documentado en https://www.tensorflow.org/rendimiento/xla/operación_semantics#pad. |
Recv <T extiende TType > | Recibe el tensor nombrado de otro cálculo XLA. |
ID de réplica | Identificación de réplica. |
SelfAdjointEig <T extiende TType > | Calcula la descomposición propia de un lote de matrices autoadjuntas (Nota: solo se admiten entradas reales). |
Enviar | Envía el tensor nombrado a otro cálculo XLA. |
Fragmentación <T extiende TType > | Una operación que fragmenta la entrada según el atributo de fragmentación dado. |
Ordenar <T extiende TType > | Envuelve el operador XLA Sort, documentado en https://www.tensorflow.org/rendimiento/xla/operación_semantics#sort. |
Svd <T extiende TType > | Calcula la descomposición propia de un lote de matrices autoadjuntas (Nota: solo se admiten entradas reales). |
XlaRecvFromHost <T extiende TType > | Una operación para recibir un tensor del anfitrión. |
XlaEnviarAlHost | Una operación para enviar un tensor al host. |
XlaSetBound | Establezca un límite para el valor de entrada dado como una sugerencia para el compilador Xla, devuelve el mismo valor. |