!10863 add dot opt into composite.__init__.py

From: @xutianming1985
Reviewed-by: @liangchenghui,@c_34
Signed-off-by: @liangchenghui
pull/10863/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit e849b370ec

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

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