Searched refs:source_matrix (Results 1 – 2 of 2) sorted by relevance
44 bool InitNonQuantizedMatrix(const EmbeddingNetworkParams::Matrix &source_matrix, in InitNonQuantizedMatrix() argument46 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() argument67 if (source_matrix.cols != 1) { in InitNonQuantizedVector()68 TC_LOG(ERROR) << "wrong #cols " << source_matrix.cols; in InitNonQuantizedVector()[all …]
52 explicit EmbeddingMatrix(const EmbeddingNetworkParams::Matrix source_matrix) in EmbeddingMatrix() argument53 : rows_(source_matrix.rows), in EmbeddingMatrix()54 cols_(source_matrix.cols), in EmbeddingMatrix()55 quant_type_(source_matrix.quant_type), in EmbeddingMatrix()56 data_(source_matrix.elements), in EmbeddingMatrix()58 quant_scales_(source_matrix.quant_scales) {} in EmbeddingMatrix()