From e691c11f30cbffb8a0334e7157191286cd8cf7bf Mon Sep 17 00:00:00 2001 From: liuxiao93 Date: Wed, 8 Jul 2020 14:41:22 +0800 Subject: [PATCH] MatMul->MatMulV2 --- mindspore/ccsrc/transform/convert.cc | 2 +- mindspore/ccsrc/transform/op_declare.cc | 10 +++++----- mindspore/ccsrc/transform/op_declare.h | 4 ++-- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/mindspore/ccsrc/transform/convert.cc b/mindspore/ccsrc/transform/convert.cc index f88e31fcd2..03b71f0ff6 100644 --- a/mindspore/ccsrc/transform/convert.cc +++ b/mindspore/ccsrc/transform/convert.cc @@ -378,7 +378,7 @@ std::unordered_map &DfGraphConvertor::get_adpt_ma {string(kNameBiasAdd), ADPT_DESC(BiasAdd)}, {prim::kPrimRelu->name(), ADPT_DESC(Relu)}, - {prim::kPrimMatMul->name(), ADPT_DESC(MatMul)}, + {prim::kPrimMatMul->name(), ADPT_DESC(MatMulV2)}, {string(kNameConst), ADPT_DESC(Constant, Const)}, {string(kNameSoftmax), ADPT_DESC(SoftmaxV2)}, diff --git a/mindspore/ccsrc/transform/op_declare.cc b/mindspore/ccsrc/transform/op_declare.cc index fd8ce624a9..bb57a630fe 100644 --- a/mindspore/ccsrc/transform/op_declare.cc +++ b/mindspore/ccsrc/transform/op_declare.cc @@ -808,11 +808,11 @@ ATTR_MAP(DepthwiseConv2DBackpropFilterD) = { }; OUTPUT_MAP(DepthwiseConv2DBackpropFilterD) = {{0, OUTPUT_DESC(filter_grad)}}; -// MatMul -INPUT_MAP(MatMul) = {{1, INPUT_DESC(x1)}, {2, INPUT_DESC(x2)}}; -ATTR_MAP(MatMul) = {{"transpose_a", ATTR_DESC(transpose_x1, AnyTraits())}, - {"transpose_b", ATTR_DESC(transpose_x2, AnyTraits())}}; -OUTPUT_MAP(MatMul) = {{0, OUTPUT_DESC(y)}}; +// MatMulV2 +INPUT_MAP(MatMulV2) = {{1, INPUT_DESC(x1)}, {2, INPUT_DESC(x2)}}; +ATTR_MAP(MatMulV2) = {{"transpose_a", ATTR_DESC(transpose_x1, AnyTraits())}, + {"transpose_b", ATTR_DESC(transpose_x2, AnyTraits())}}; +OUTPUT_MAP(MatMulV2) = {{0, OUTPUT_DESC(y)}}; // Merge INPUT_MAP(Merge) = EMPTY_INPUT_MAP; diff --git a/mindspore/ccsrc/transform/op_declare.h b/mindspore/ccsrc/transform/op_declare.h index baa819f71f..c329e60454 100755 --- a/mindspore/ccsrc/transform/op_declare.h +++ b/mindspore/ccsrc/transform/op_declare.h @@ -313,8 +313,8 @@ DECLARE_OP_ADAPTER(NPUAllocFloatStatus) DECLARE_OP_USE_OUTPUT(NPUAllocFloatStatus) DECLARE_OP_ADAPTER(NPUClearFloatStatus) DECLARE_OP_USE_OUTPUT(NPUClearFloatStatus) -DECLARE_OP_ADAPTER(MatMul) -DECLARE_OP_USE_OUTPUT(MatMul) +DECLARE_OP_ADAPTER(MatMulV2) +DECLARE_OP_USE_OUTPUT(MatMulV2) DECLARE_OP_ADAPTER(SoftmaxCrossEntropyWithLogits) DECLARE_OP_USE_OUTPUT(SoftmaxCrossEntropyWithLogits)