Searched refs:embeddings_size (Results 1 – 4 of 4) sorted by relevance
| /external/libtextclassifier/common/memory_image/ |
| D | embedding-network-params-from-image.h | 43 hidden_blob_offset_ = embeddings_blob_offset_ + embeddings_size(); in EmbeddingNetworkParamsFromImage()
44 if (trimmed_proto_.embeddings_size() && in EmbeddingNetworkParamsFromImage()
48 hidden_blob_offset_ += embeddings_size(); in EmbeddingNetworkParamsFromImage()
68 int embeddings_size() const override { in embeddings_size() function
69 return trimmed_proto_.embeddings_size(); in embeddings_size()
73 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_num_rows()
78 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_num_cols()
83 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_weights()
92 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_quant_type()
101 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_quant_scales()
|
| /external/libtextclassifier/common/ |
| D | embedding-network-params-from-proto.h | 60 embeddings_quant_scales_.resize(proto_->embeddings_size()); in EmbeddingNetworkParamsFromProto()
61 embeddings_quant_weights_.resize(proto_->embeddings_size()); in EmbeddingNetworkParamsFromProto()
62 for (int i = 0; i < proto_->embeddings_size(); ++i) { in EmbeddingNetworkParamsFromProto()
113 int embeddings_size() const override { return proto_->embeddings_size(); } in embeddings_size() function
116 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_num_rows()
121 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_num_cols()
126 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_weights()
135 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_quant_type()
141 TC_DCHECK(InRange(i, embeddings_size())); in embeddings_quant_scales()
|
| D | embedding-network-params.h | 66 if (embeddings_size() != embedding_num_features_size()) { in GetNumEmbeddingSpaces()
67 TC_LOG(ERROR) << "Embedding spaces mismatch " << embeddings_size() in GetNumEmbeddingSpaces()
71 std::min(embeddings_size(), embedding_num_features_size())); in GetNumEmbeddingSpaces()
79 TC_DCHECK(InRange(i, embeddings_size())); in GetEmbeddingMatrix()
224 virtual int embeddings_size() const = 0;
|
| /external/libtextclassifier/smartselect/ |
| D | model-params.h | 34 int embeddings_size() const override { return context_size_ * 2 + 1; } in embeddings_size() function
90 int embeddings_size() const override { in embeddings_size() function
91 return embedding_params_->embeddings_size(); in embeddings_size()
|