Lines Matching refs:source_matrix
44 bool InitNonQuantizedMatrix(const EmbeddingNetworkParams::Matrix &source_matrix, in InitNonQuantizedMatrix() argument
46 mat->resize(source_matrix.rows); in InitNonQuantizedMatrix()
50 if (!CheckNoQuantization(source_matrix)) return false; in InitNonQuantizedMatrix()
52 reinterpret_cast<const float *>(source_matrix.elements); in InitNonQuantizedMatrix()
53 for (int r = 0; r < source_matrix.rows; ++r) { in InitNonQuantizedMatrix()
54 (*mat)[r] = EmbeddingNetwork::VectorWrapper(weights, source_matrix.cols); in InitNonQuantizedMatrix()
55 weights += source_matrix.cols; in InitNonQuantizedMatrix()
65 bool InitNonQuantizedVector(const EmbeddingNetworkParams::Matrix &source_matrix, in InitNonQuantizedVector() argument
67 if (source_matrix.cols != 1) { in InitNonQuantizedVector()
68 TC_LOG(ERROR) << "wrong #cols " << source_matrix.cols; in InitNonQuantizedVector()
71 if (!CheckNoQuantization(source_matrix)) { in InitNonQuantizedVector()
77 if (!CheckNoQuantization(source_matrix)) return false; in InitNonQuantizedVector()
79 reinterpret_cast<const float *>(source_matrix.elements); in InitNonQuantizedVector()
80 *vector = EmbeddingNetwork::VectorWrapper(weights, source_matrix.rows); in InitNonQuantizedVector()