XlaSparseDenseMatmul

lezione finale pubblica XlaSparseDenseMatmul

Metodi pubblici

Uscita <Mobile>
statico XlaSparseDenseMatmul
create ( ambito ambito , operando <intero> rowId, operando <?> colId, valori operando <float>, offset operando <?>, operando <float> embeddingTable, lungo maxIdsPerPartition, lungo maxUniqueIdsPerPartition, lungo inputSize)
Metodo factory per creare una classe che racchiude una nuova operazione XlaSparseDenseMatmul.
Output <Intero>
Output <Intero>
Uscita <Mobile>
Output <Intero>

Metodi ereditati

Metodi pubblici

attivazioni Output <Float> pubbliche ()

public static XlaSparseDenseMatmul create ( scope scope, operando <Integer> rowIds, operando <?> colIds, operando <Float> valori, operando <?> offset, operando <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

Metodo factory per creare una classe che racchiude una nuova operazione XlaSparseDenseMatmul.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di XlaSparseDenseMatmul

output pubblico <intero> rowPointers ()

output pubblico <intero> sortedEmbeddingIds ()

output pubblico <Float> sortedGains ()

output pubblico <intero> sortedSampleIds ()