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