Browse Source

NotEqual op auto cast

Signed-off-by: candanzg <zhangshucheng@huawei.com>
tags/v0.2.0-alpha
candanzg 5 years ago
parent
commit
79c4312e89
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

@@ -401,6 +401,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