Home
last modified time | relevance | path

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

/external/libtextclassifier/common/
Dembedding-network.h128 class VectorWrapper {
130 VectorWrapper() : VectorWrapper(nullptr, 0) {} in VectorWrapper() function
136 VectorWrapper(const float *data, int size) : data_(data), size_(size) {} in VectorWrapper() function
149 typedef std::vector<VectorWrapper> Matrix;
236 std::vector<VectorWrapper> hidden_bias_;
240 VectorWrapper softmax_bias_;
Dembedding-network.cc54 (*mat)[r] = EmbeddingNetwork::VectorWrapper(weights, source_matrix.cols); in InitNonQuantizedMatrix()
66 EmbeddingNetwork::VectorWrapper *vector) { in InitNonQuantizedVector()
80 *vector = EmbeddingNetwork::VectorWrapper(weights, source_matrix.rows); in InitNonQuantizedVector()
88 const EmbeddingNetwork::VectorWrapper &b, in SparseReluProductPlusBias()