HomeSort by relevance Sort by last modified time
    Searched refs:selection_network_ (Results 1 - 2 of 2) sorted by null

  /external/libtextclassifier/smartselect/
text-classification-model.cc 188 selection_network_.reset(new EmbeddingNetwork(selection_params_.get()));
192 *selection_network_, selection_network_->EmbeddingSize(0));
349 *selection_network_, selection_feature_fn_, &selection_label_spans);
373 int embedding_size = selection_network_->EmbeddingSize(0);
398 selection_network_->ComputeLogits(features, &scores);
text-classification-model.h 115 std::unique_ptr<nlp_core::EmbeddingNetwork> selection_network_; member in class:libtextclassifier::TextClassificationModel

Completed in 173 milliseconds