From 37f185b12bef42c95c06f33ac29138e3b851e36e Mon Sep 17 00:00:00 2001 From: xutianming Date: Thu, 31 Dec 2020 09:33:20 +0800 Subject: [PATCH] add dot opt into composite.__init__.py --- mindspore/ops/composite/__init__.py | 3 ++- tests/st/ops/cpu/test_dot_op.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/ops/composite/__init__.py b/mindspore/ops/composite/__init__.py index eda1ccada7..e9c09c2c14 100644 --- a/mindspore/ops/composite/__init__.py +++ b/mindspore/ops/composite/__init__.py @@ -27,7 +27,7 @@ from .multitype_ops.add_impl import hyper_add from .multitype_ops.ones_like_impl import ones_like from .multitype_ops.zeros_like_impl import zeros_like from .random_ops import normal, laplace, uniform, gamma, poisson, multinomial -from .math_ops import count_nonzero, tensor_dot, batch_dot +from .math_ops import count_nonzero, tensor_dot, dot, batch_dot from .array_ops import repeat_elements, sequence_mask @@ -53,6 +53,7 @@ __all__ = [ 'clip_by_global_norm', 'count_nonzero', 'tensor_dot', + 'dot', 'batch_dot', 'repeat_elements', 'sequence_mask'] diff --git a/tests/st/ops/cpu/test_dot_op.py b/tests/st/ops/cpu/test_dot_op.py index cb17725bdf..a3c68889a7 100644 --- a/tests/st/ops/cpu/test_dot_op.py +++ b/tests/st/ops/cpu/test_dot_op.py @@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="CPU") class NetDot(nn.Cell): def construct(self, x, y): - return C.math_ops.dot(x, y) + return C.dot(x, y) @pytest.mark.level0