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