Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!40760 [MS][lite] kronecker production decomposition warning fixed 

Merge pull request !40760 from lz/pr_40736
This commit is contained in:
i-robot 2022-08-27 08:26:24 +00:00 committed by Gitee
parent 8196e1a3f4 3f0e2188d7
commit e22c122959
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/lite/tools/converter/decomposer/kp_decomposition.h
View File

@ -30,10 +30,10 @@ class KPDecomposition {
int Decomposition();
std::vector<float> GetMatB() { return mat_b_; }
std::vector<int> GetMatShapeB() { return mat_shape_b_; }
std::vector<float> GetMatC() { return mat_c_; }
std::vector<int> GetMatShapeC() { return mat_shape_c_; }
std::vector<float> GetMatB() const { return mat_b_; }
std::vector<int> GetMatShapeB() const { return mat_shape_b_; }
std::vector<float> GetMatC() const { return mat_c_; }
std::vector<int> GetMatShapeC() const { return mat_shape_c_; }
private:
// Integer factorization to multiply two constraints ==> A = B * C