org.tensorflow.op.xla

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.