Home
last modified time | relevance | path

Searched refs:hidden_weights_ (Results 1 – 2 of 2) sorted by relevance

/external/libtextclassifier/common/
Dembedding-network.cc261 for (int i = 0; i < hidden_weights_.size(); ++i) { in FinishComputeFinalScoresInternal()
266 apply_relu, hidden_weights_[i], hidden_bias_[i], h_in, h_out)) { in FinishComputeFinalScoresInternal()
345 hidden_weights_.resize(num_hidden_layers); in EmbeddingNetwork()
352 if (!InitNonQuantizedMatrix(matrix, &hidden_weights_[i]) || in EmbeddingNetwork()
Dembedding-network.h235 std::vector<Matrix> hidden_weights_; variable