Browse Source

!456 Auto cast case for NotEqual op

Merge pull request !456 from candanzg/not_equal_auto_cast
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
27a88a6bc3
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      tests/ut/python/ops/test_ops.py

+ 5
- 0
tests/ut/python/ops/test_ops.py View File

@@ -414,6 +414,11 @@ test_case_math_ops = [
'block': P.NotEqual(),
'desc_inputs': [[4, 1], [2, 3, 4, 5]],
'desc_bprop': [Tensor(np.ones((2, 3, 4, 5), np.bool_))]}),
('NotEqual_0', {
'block': P.NotEqual(),
'desc_inputs': [ 1, [2, 3, 4, 5]],
'desc_bprop': [Tensor(np.ones((2, 3, 4, 5), np.bool_))],
'skip': ['backward']}),
('Greater', {
'block': P.Greater(),
'desc_inputs': [[2, 3, 4, 1], [4, 5]],


Loading…
Cancel
Save