XlaSparseDenseMatmul

classe final pública XlaSparseDenseMatmul

Métodos Públicos

Saída <flutuante>
XlaSparseDenseMatmul estático
create (Escopo do escopo , Operando <Integer> rowIds, Operando <?> colIds, Operando <Float> valores, Operando <?> deslocamentos, Operando <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
Método de fábrica para criar uma classe que envolve uma nova operação XlaSparseDenseMatmul.
Saída <Inteiro>
Saída <Inteiro>
Saída <flutuante>
Saída <Inteiro>

Métodos herdados

Métodos Públicos

ativações de saída pública <Float> ()

public static XlaSparseDenseMatmul create ( Escopo de escopo , Operando <Integer> rowIds, Operando <?> colIds, Operando <Float> valores, Operando <?> deslocamentos, Operando <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

Método de fábrica para criar uma classe que envolve uma nova operação XlaSparseDenseMatmul.

Parâmetros
escopo escopo atual
Devoluções
  • uma nova instância de XlaSparseDenseMatmul

Saída pública <Inteiro> rowPointers ()

saída pública <inteiro> sortedEmbeddingIds ()

Saída pública <Float> sortedGains ()

Saída pública <Inteiro> sortedSampleIds ()