org.tensorflow.op.xla

Aulas

BroadcastHelper <T extends TType > Operador auxiliar para realizar transmissões no estilo XLA

Transmite `lhs` e` rhs` para a mesma classificação, adicionando dimensões de tamanho 1 para qualquer um de `lhs` e` rhs` que tiver a classificação mais baixa, usando as regras de transmissão do XLA para operadores binários.

ClusterOutput <T extends TType > Operador que conecta a saída de um cálculo XLA a outros nós de gráfico do consumidor.
Conv <T extends TType > Envolve o operador XLA ConvGeneralDilated, documentado em

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

Desquantizar Pega a entrada uint32 compactada e descompacta a entrada para uint8 para fazer

Desquantização no dispositivo.

Ponto <T estende TType > Envolve o operador XLA DotGeneral, documentado em

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

DynamicSlice <T extends TType > Envolve o operador XLA DynamicSlice, documentado em

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

DynamicUpdateSlice <T extends TType > Envolve o operador XLA DynamicUpdateSlice, documentado em

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

Einsum <T extends TType > Um op que suporta op einsum básico com 2 entradas e 1 saída.
Reúna <T extends TType > Envolve o operador XLA Gather documentado em

https://www.tensorflow.org/xla/operation_semantics#gather

KeyValueSort <T estende TNumber , U estende TType > Envolve o operador XLA Sort, documentado em

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

Pad <T extends TType > Envolve o operador do XLA Pad, documentado em

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

Recv <T extends TType > Recebe o tensor nomeado de outro cálculo XLA.
ReplicaId ID da réplica.
SelfAdjointEig <T extends TType > Calcula a decomposição própria de um lote de matrizes auto-adjuntas

(Nota: apenas entradas reais são suportadas).

Mandar Envia o tensor nomeado para outro cálculo XLA.
Fragmentação <T extends TType > Uma operação que fragmenta a entrada com base no atributo de fragmentação fornecido.
Classificar <T extends TType > Envolve o operador XLA Sort, documentado em

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

Svd <T extends TType > Calcula a decomposição própria de um lote de matrizes auto-adjuntas

(Nota: apenas entradas reais são suportadas).

XlaRecvFromHost <T extends TType > Um op para receber um tensor do host.
XlaSendToHost Uma op para enviar um tensor ao host.
XlaSetBound Defina um limite para o valor de entrada fornecido como uma dica para o compilador Xla,

retorna o mesmo valor.