diff --git a/graphengine b/graphengine index 103f2d1019..2cb83c8f4d 160000 --- a/graphengine +++ b/graphengine @@ -1 +1 @@ -Subproject commit 103f2d1019dc50d781d7a964551d9f1f50b3b009 +Subproject commit 2cb83c8f4d8d3799081d016dd453a459e25c9879 diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare.cc b/mindspore/ccsrc/transform/graph_ir/op_declare.cc index 1f803d4d81..4e48ada9fd 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare.cc +++ b/mindspore/ccsrc/transform/graph_ir/op_declare.cc @@ -309,12 +309,7 @@ INPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{1, INPUT_DESC(features)}, {2, INPUT ATTR_MAP(SoftmaxCrossEntropyWithLogits) = EMPTY_ATTR_MAP; OUTPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{0, OUTPUT_DESC(loss)}, {1, OUTPUT_DESC(backprop)}}; -// MeanGrad -INPUT_MAP(MeanGrad) = {{1, INPUT_DESC(x)}}; -INPUT_ATTR_MAP(MeanGrad) = {{2, ATTR_DESC(mean_grad_output_shape_value, kOpFormat_NHWC, - AnyTraits>(), AnyTraits())}}; -ATTR_MAP(MeanGrad) = {{"mode", ATTR_DESC(mode, AnyTraits())}}; - +// SliceD INPUT_MAP(SliceD) = {{1, INPUT_DESC(x)}}; INPUT_ATTR_MAP(SliceD) = {{2, ATTR_DESC(offsets, AnyTraits(), AnyTraits>())}, {3, ATTR_DESC(size, AnyTraits(), AnyTraits>())}}; @@ -431,11 +426,6 @@ INPUT_MAP(TopK) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(k)}}; ATTR_MAP(TopK) = {{"sorted", ATTR_DESC(sorted, AnyTraits())}}; OUTPUT_MAP(TopK) = {{0, OUTPUT_DESC(values)}, {1, OUTPUT_DESC(indices)}}; -// Multiply -INPUT_MAP(Multiply) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(y)}}; -ATTR_MAP(Multiply) = EMPTY_ATTR_MAP; -OUTPUT_MAP(Multiply) = {{0, OUTPUT_DESC(z)}}; - // TileD INPUT_MAP(TileD) = {{1, INPUT_DESC(x)}}; INPUT_ATTR_MAP(TileD) = {{2, ATTR_DESC(multiples, AnyTraits(), AnyTraits>())}}; diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare.h b/mindspore/ccsrc/transform/graph_ir/op_declare.h index 220cc945a7..0b32bf8195 100755 --- a/mindspore/ccsrc/transform/graph_ir/op_declare.h +++ b/mindspore/ccsrc/transform/graph_ir/op_declare.h @@ -70,8 +70,6 @@ DECLARE_OP_ADAPTER(AssignSub) DECLARE_OP_USE_OUTPUT(AssignSub) DECLARE_OP_ADAPTER(ReduceMean) -DECLARE_OP_ADAPTER(Multiply) -DECLARE_OP_USE_OUTPUT(Multiply) // ** Distributed Operations ** DECLARE_OP_ADAPTER(HcomReduceScatter) @@ -327,9 +325,6 @@ DECLARE_OP_USE_OUTPUT(MatMulV2) DECLARE_OP_ADAPTER(SoftmaxCrossEntropyWithLogits) DECLARE_OP_USE_OUTPUT(SoftmaxCrossEntropyWithLogits) -DECLARE_OP_ADAPTER(MeanGrad) -DECLARE_OP_USE_INPUT_ATTR(MeanGrad) - DECLARE_OP_ADAPTER(Assign) DECLARE_OP_USE_OUTPUT(Assign) DECLARE_OP_ADAPTER(Constant) diff --git a/mindspore/ccsrc/transform/op_declare.cc b/mindspore/ccsrc/transform/op_declare.cc index 9d86b30b17..3a510fc3e5 100644 --- a/mindspore/ccsrc/transform/op_declare.cc +++ b/mindspore/ccsrc/transform/op_declare.cc @@ -293,12 +293,7 @@ INPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{1, INPUT_DESC(features)}, {2, INPUT ATTR_MAP(SoftmaxCrossEntropyWithLogits) = EMPTY_ATTR_MAP; OUTPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{0, OUTPUT_DESC(loss)}, {1, OUTPUT_DESC(backprop)}}; -// MeanGrad -INPUT_MAP(MeanGrad) = {{1, INPUT_DESC(x)}}; -INPUT_ATTR_MAP(MeanGrad) = {{2, ATTR_DESC(mean_grad_output_shape_value, kOpFormat_NHWC, - AnyTraits>(), AnyTraits())}}; -ATTR_MAP(MeanGrad) = {{"mode", ATTR_DESC(mode, AnyTraits())}}; - +// SliceD INPUT_MAP(SliceD) = {{1, INPUT_DESC(x)}}; INPUT_ATTR_MAP(SliceD) = {{2, ATTR_DESC(offsets, AnyTraits(), AnyTraits>())}, {3, ATTR_DESC(size, AnyTraits(), AnyTraits>())}}; @@ -415,11 +410,6 @@ INPUT_MAP(TopK) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(k)}}; ATTR_MAP(TopK) = {{"sorted", ATTR_DESC(sorted, AnyTraits())}}; OUTPUT_MAP(TopK) = {{0, OUTPUT_DESC(values)}, {1, OUTPUT_DESC(indices)}}; -// Multiply -INPUT_MAP(Multiply) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(y)}}; -ATTR_MAP(Multiply) = EMPTY_ATTR_MAP; -OUTPUT_MAP(Multiply) = {{0, OUTPUT_DESC(z)}}; - // TileD INPUT_MAP(TileD) = {{1, INPUT_DESC(x)}}; INPUT_ATTR_MAP(TileD) = {{2, ATTR_DESC(multiples, AnyTraits(), AnyTraits>())}};