Home
last modified time | relevance | path

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

/external/libtextclassifier/common/memory_image/
Dembedding-network-params-from-image.h40 trimmed_proto_(memory_reader_.trimmed_proto()) { in EmbeddingNetworkParamsFromImage()
44 if (trimmed_proto_.embeddings_size() && in EmbeddingNetworkParamsFromImage()
45 trimmed_proto_.embeddings(0).is_quantized()) { in EmbeddingNetworkParamsFromImage()
60 if (trimmed_proto_.HasExtension(extension_id)) { in GetTaskSpec()
61 return &(trimmed_proto_.GetExtension(extension_id)); in GetTaskSpec()
69 return trimmed_proto_.embeddings_size(); in embeddings_size()
74 return trimmed_proto_.embeddings(i).rows(); in embeddings_num_rows()
79 return trimmed_proto_.embeddings(i).cols(); in embeddings_num_cols()
84 const int blob_index = trimmed_proto_.embeddings(i).is_quantized() in embeddings_weights()
93 if (trimmed_proto_.embeddings(i).is_quantized()) { in embeddings_quant_type()
[all …]
Dmemory-image-reader.h128 if (!trimmed_proto_.ParseFromString(trimmed_proto_str())) { in MemoryImageReader()
137 const T &trimmed_proto() const { return trimmed_proto_; } in trimmed_proto()
140 T trimmed_proto_;