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

!49184 fix matrix adapter mapping name 

Merge pull request !49184 from qiuzhongya/matrix
This commit is contained in:
i-robot 2023-02-22 01:06:16 +00:00 committed by Gitee
parent 552fbb71dc fbc8f469bd
commit 753fbf51c1
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/transform/graph_ir/op_adapter_map.h
View File

@ -181,8 +181,8 @@ constexpr const char kNameLRNGrad[] = "LRNGrad";
constexpr const char kNameLSTMInputGrad[] = "LSTMInputGrad";
constexpr const char kNameMatMul[] = "MatMul";
constexpr const char kNameMatrixDiag[] = "MatrixDiag";
constexpr const char kNameMatrixDiagPartD[] = "MatrixDiagPart";
constexpr const char kNameMatrixSetDiagD[] = "MatrixSetDiag";
constexpr const char kNameMatrixDiagPartD[] = "MatrixDiagPartD";
constexpr const char kNameMatrixSetDiagD[] = "MatrixSetDiagD";
constexpr const char kNameMaxPool3DGradGrad[] = "MaxPool3DGradGrad";
constexpr const char kNameMaxPoolGradGrad[] = "MaxPoolGradGrad";
constexpr const char kNameMaxPoolGradGradWithArgmax[] = "MaxPoolGradGradWithArgmax";

4
mindspore/ccsrc/transform/graph_ir/op_declare/matrix_calculation_ops_declare.cc
View File

@ -122,13 +122,13 @@ REG_ADPT_DESC(MatrixSetDiagD, kNameMatrixSetDiagD, ADPT_DESC(MatrixSetDiagD))
INPUT_MAP(MatrixDiagPart) = {{1, INPUT_DESC(x)}};
ATTR_MAP(MatrixDiagPart) = EMPTY_ATTR_MAP;
OUTPUT_MAP(MatrixDiagPart) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(MatrixDiagPart, kMatrixDiagPartDOpName, ADPT_DESC(MatrixDiagPart))
REG_ADPT_DESC(MatrixDiagPart, kMatrixDiagPartOpName, ADPT_DESC(MatrixDiagPart))
// MatrixSetDiag
INPUT_MAP(MatrixSetDiag) = {{1, INPUT_DESC(x)}};
ATTR_MAP(MatrixSetDiag) = EMPTY_ATTR_MAP;
OUTPUT_MAP(MatrixSetDiag) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(MatrixSetDiag, kMatrixSetDiagDOpName, ADPT_DESC(MatrixSetDiag))
REG_ADPT_DESC(MatrixSetDiag, kMatrixSetDiagOpName, ADPT_DESC(MatrixSetDiag))
// ConfusionMatrix
INPUT_MAP(ConfusionMatrix) = {{1, INPUT_DESC(labels)}, {2, INPUT_DESC(predictions)}, {3, INPUT_DESC(weights)}};