org.tensorflow.op.xla

Clases

BroadcastHelper <T extiende TType > Operador auxiliar para realizar transmisiones al estilo XLA

Transmite `lhs` y` rhs` al mismo rango, agregando dimensiones de tamaño 1 a cualquiera de `lhs` y` rhs` que tenga el rango más bajo, utilizando las reglas de transmisión de XLA para operadores binarios.

ClusterOutput <T extiende TType > Operador que conecta la salida de un cálculo XLA a otros nodos de gráficos de consumidores.
Conv <T extiende TType > Envuelve el operador XLA ConvGeneralDilated, documentado en

https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution.

Descuantizar Toma la entrada empaquetada de uint32 y descomprime la entrada a uint8 para hacer

Descuantización en dispositivo.

Punto <T extiende TType > Envuelve el operador XLA DotGeneral, documentado en

https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral.

DynamicSlice <T extiende TType > Envuelve el operador XLA DynamicSlice, documentado en

https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice.

DynamicUpdateSlice <T extiende TType > Envuelve el operador XLA DynamicUpdateSlice, documentado en

https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice.

Einsum <T extiende TType > Una operación que admite operaciones básicas de einsum con 2 entradas y 1 salida.
Reúna <T extiende TType > Envuelve el operador de 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/performance/xla/operation_semantics#sort.

Pad <T extiende TType > Envuelve el operador de XLA Pad, documentado en

https://www.tensorflow.org/performance/xla/operation_semantics#pad.

Recv <T extiende TType > Recibe el tensor nombrado de otro cálculo XLA.
ReplicaId ID 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/performance/xla/operation_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.
XlaSendToHost Una operación para enviar un tensor al anfitrión.
XlaSetBound Establezca un límite para el valor de entrada dado como una sugerencia para el compilador Xla,

devuelve el mismo valor.