Browse Source

add dot opt into composite.__init__.py

tags/v1.2.0-rc1
xutianming 4 years ago
parent
commit
37f185b12b
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      mindspore/ops/composite/__init__.py
  2. +1
    -1
      tests/st/ops/cpu/test_dot_op.py

+ 2
- 1
mindspore/ops/composite/__init__.py View File

@@ -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']

+ 1
- 1
tests/st/ops/cpu/test_dot_op.py View File

@@ -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


Loading…
Cancel
Save