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. |