org.tensorflow.op.xla

Aulas

BroadcastHelper <T estende 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 a qualquer uma das `lhs` e `rhs` que tenha a classificação mais baixa, usando as regras de transmissão do XLA para operadores binários.

ClusterOutput <T estende TType > Operador que conecta a saída de uma computação XLA a outros nós gráficos consumidores.
Conv <T estende 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 estende TType > Envolve o operador XLA DynamicSlice, documentado em

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

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

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

Einsum <T estende TType > Uma operação que suporta operação einsum básica com 2 entradas e 1 saída.
Reunir <T estende 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 estende TType > Envolve o operador XLA Pad, documentado em

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

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

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

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

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

Svd <T estende TType > Calcula a decomposição própria de um lote de matrizes autoadjuntas

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

XlaRecvFromHost <T estende TType > Uma operação para receber um tensor do host.
XlaSendToHost Uma operação 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.