!456 Auto cast case for NotEqual op

Merge pull request !456 from candanzg/not_equal_auto_cast
2020-04-21 09:20:40 +08:00 · 2020-04-21 09:20:40 +08:00 · 27a88a6bc3
parent 717ed427b2 79c4312e89
commit 27a88a6bc3
1 changed files with 5 additions and 0 deletions
--- a/tests/ut/python/ops/test_ops.py
+++ b/tests/ut/python/ops/test_ops.py
@ -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]],