Searched refs:selection_network_ (Results 1 – 2 of 2) sorted by relevance
188 selection_network_.reset(new EmbeddingNetwork(selection_params_.get())); in LoadModels()192 *selection_network_, selection_network_->EmbeddingSize(0)); in LoadModels()349 *selection_network_, selection_feature_fn_, &selection_label_spans); in SuggestSelectionInternal()373 int embedding_size = selection_network_->EmbeddingSize(0); in SuggestSelectionSymmetrical()398 selection_network_->ComputeLogits(features, &scores); in SuggestSelectionSymmetrical()
115 std::unique_ptr<nlp_core::EmbeddingNetwork> selection_network_; variable