Lines Matching refs:Matrix
30 // Returns true if and only if matrix does not use any quantization.
31 bool CheckNoQuantization(const EmbeddingNetworkParams::Matrix &matrix) {
32 if (matrix.quant_type != QuantizationType::NONE) {
40 // Initializes a Matrix object with the parameters from the MatrixParams
44 bool InitNonQuantizedMatrix(const EmbeddingNetworkParams::Matrix &source_matrix,
45 EmbeddingNetwork::Matrix *mat) {
65 bool InitNonQuantizedVector(const EmbeddingNetworkParams::Matrix &source_matrix,
87 const EmbeddingNetwork::Matrix &weights,
317 const EmbeddingNetworkParams::Matrix matrix = model->GetEmbeddingMatrix(i);
318 if (matrix.quant_type != QuantizationType::UINT8) {
320 << static_cast<int>(matrix.quant_type);
324 // There is no way to accomodate an empty embedding matrix. E.g., there is
327 if (matrix.rows == 0) {
328 TC_LOG(ERROR) << "Empty embedding matrix #" << i;
331 embedding_matrices_.emplace_back(new EmbeddingMatrix(matrix));
349 const EmbeddingNetworkParams::Matrix matrix =
351 const EmbeddingNetworkParams::Matrix bias = model->GetHiddenLayerBias(i);
352 if (!InitNonQuantizedMatrix(matrix, &hidden_weights_[i]) ||
363 const EmbeddingNetworkParams::Matrix softmax = model->GetSoftmaxMatrix();
364 const EmbeddingNetworkParams::Matrix softmax_bias = model->GetSoftmaxBias();
