Searched refs:layer_weights_ (Results 1 – 2 of 2) sorted by relevance
270 const int num_layers = layer_weights_.size(); in ComputeFinalScores()283 apply_relu, layer_weights_[i], layer_bias_[i], *v_in, v_out); in ComputeFinalScores()305 layer_weights_.push_back(model_->GetHiddenLayerMatrix(i)); in EmbeddingNetwork()314 layer_weights_.push_back(model_->GetSoftmaxMatrix()); in EmbeddingNetwork()
92 std::vector<EmbeddingNetworkParams::Matrix> layer_weights_; variable