Searched refs:EmbeddingNetwork (Results 1 – 6 of 6) sorted by relevance
43 class EmbeddingNetwork {49 explicit EmbeddingNetwork(const EmbeddingNetworkParams *model);51 virtual ~EmbeddingNetwork() {} in ~EmbeddingNetwork()
148 void EmbeddingNetwork::ConcatEmbeddings( in ConcatEmbeddings()244 void EmbeddingNetwork::ComputeFinalScores( in ComputeFinalScores()250 void EmbeddingNetwork::ComputeFinalScores( in ComputeFinalScores()288 EmbeddingNetwork::EmbeddingNetwork(const EmbeddingNetworkParams *model) in EmbeddingNetwork() function in libtextclassifier3::mobile::EmbeddingNetwork
79 network_.reset(new EmbeddingNetwork(nn_params)); in LangIdImpl()272 std::unique_ptr<EmbeddingNetwork> network_;
109 table EmbeddingNetwork {117 root_type EmbeddingNetwork;
279 const saft_fbs::EmbeddingNetwork *network_ = nullptr;
57 // the input bytes., e.g., "EmbeddingNetwork". If |type| is proto, this