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

  /external/libtextclassifier/lang_id/common/
embedding-network.h 32 // Based on the more general nlp_saft::EmbeddingNetwork (without ::mobile).
43 class EmbeddingNetwork {
48 // EmbeddingNetwork object.
49 explicit EmbeddingNetwork(const EmbeddingNetworkParams *model);
51 virtual ~EmbeddingNetwork() {}
embedding-network.cc 72 // CHECK-ed that the EmbeddingNetwork constructor.
148 void EmbeddingNetwork::ConcatEmbeddings(
244 void EmbeddingNetwork::ComputeFinalScores(
250 void EmbeddingNetwork::ComputeFinalScores(
288 EmbeddingNetwork::EmbeddingNetwork(const EmbeddingNetworkParams *model)
  /external/libtextclassifier/lang_id/
lang-id.cc 74 network_.reset(new EmbeddingNetwork(nn_params));
238 std::unique_ptr<EmbeddingNetwork> network_;
  /external/libtextclassifier/lang_id/common/flatbuffers/
embedding-network-params-from-flatbuffer.h 277 // EmbeddingNetwork flatbuffer from the bytes passed as parameter to the
279 const saft_fbs::EmbeddingNetwork *network_ = nullptr;

Completed in 1596 milliseconds