Home
last modified time | relevance | path

Searched refs:EmbeddingParams (Results 1 – 2 of 2) sorted by relevance

/external/libtextclassifier/smartselect/
Dmodel-params.h28 class EmbeddingParams : public nlp_core::EmbeddingNetworkParamsFromImage {
30 EmbeddingParams(const void* start, uint64 num_bytes, int context_size) in EmbeddingParams() function
85 std::shared_ptr<EmbeddingParams> GetEmbeddingParams() const { in GetEmbeddingParams()
125 std::shared_ptr<EmbeddingParams> external_embedding_params);
128 std::shared_ptr<EmbeddingParams> embedding_params, in ModelParams()
143 std::shared_ptr<EmbeddingParams> embedding_params_;
148 std::shared_ptr<EmbeddingParams> external_embedding_params);
Dmodel-params.cc28 std::shared_ptr<EmbeddingParams> external_embedding_params) { in ModelParamsBuilder()
121 std::shared_ptr<EmbeddingParams> embedding_params(new EmbeddingParams( in ModelParamsBuilder()