Searched refs:softmax_weights_ (Results 1 – 2 of 2) sorted by relevance
274 true, softmax_weights_, softmax_bias_, h_in, scores)) { in FinishComputeFinalScoresInternal()365 if (!InitNonQuantizedMatrix(softmax, &softmax_weights_) || in EmbeddingNetwork()
239 Matrix softmax_weights_; variable