Home
last modified time | relevance | path

Searched refs:equal_nan (Results 1 – 25 of 40) sorted by relevance

12

/third_party/mindspore/tests/ut/python/dataset/
Dtest_time_stretch.py47 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
49 assert np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan)
50 elif not np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan):
Dtest_amplitude_to_db.py48 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
51 assert np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan)
52 elif not np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan):
Dtest_frequency_masking.py48 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
51 assert np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan)
52 elif not np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan):
Dtest_time_masking.py48 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
51 assert np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan)
52 elif not np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan):
Dtest_vol.py39 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
42 assert np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan)
43 elif not np.allclose(data_me, data_expected, rtol, atol, equal_nan=equal_nan):
/third_party/mindspore/tests/st/ops/cpu/
Dtest_squeeze_op.py44 atol=1.e-8, equal_nan=True)
56 atol=1.e-8, equal_nan=True)
68 atol=1.e-8, equal_nan=True)
81 atol=1.e-8, equal_nan=True)
94 atol=1.e-8, equal_nan=True)
Dtest_maximum_grad.py47 assert np.allclose(result[0].asnumpy(), dx, rtol=1.e-4, atol=1.e-8, equal_nan=True)
48 assert np.allclose(result[1].asnumpy(), dy, rtol=1.e-4, atol=1.e-8, equal_nan=True)
169 assert np.allclose(result[0].asnumpy(), dx, rtol=1.e-4, atol=1.e-8, equal_nan=True)
170 assert np.allclose(result[1].asnumpy(), dy, rtol=1.e-4, atol=1.e-8, equal_nan=True)
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_fused_adam.py152 rtol=1.e-4, atol=1.e-8, equal_nan=True)
154 rtol=1.e-4, atol=1.e-8, equal_nan=True)
156 rtol=1.e-4, atol=1.e-8, equal_nan=True)
182 rtol=1.e-4, atol=1.e-8, equal_nan=True)
184 rtol=1.e-4, atol=1.e-8, equal_nan=True)
186 rtol=1.e-4, atol=1.e-8, equal_nan=True)
211 rtol=1.e-4, atol=1.e-8, equal_nan=True)
213 rtol=1.e-4, atol=1.e-8, equal_nan=True)
215 rtol=1.e-4, atol=1.e-8, equal_nan=True)
217 rtol=1.e-4, atol=1.e-8, equal_nan=True)
Dtest_atomic_add.py71 res = np.allclose(expect, result.asnumpy(), rtol=1.e-4, atol=1.e-7, equal_nan=True)
85 res = np.allclose(expect, result.asnumpy(), rtol=1.e-4, atol=1.e-7, equal_nan=True)
99 res = np.allclose(expect, result.asnumpy(), rtol=1.e-4, atol=1.e-7, equal_nan=True)
Dtest_maximum_grad.py55 assert np.allclose(result_on[0], result_off[0], rtol=1.e-4, atol=1.e-8, equal_nan=True)
56 assert np.allclose(result_on[1], result_off[1], rtol=1.e-4, atol=1.e-8, equal_nan=True)\
Dtest_minimum_grad.py55 assert np.allclose(result_on[0], result_off[0], rtol=1.e-4, atol=1.e-8, equal_nan=True)
56 assert np.allclose(result_on[1], result_off[1], rtol=1.e-4, atol=1.e-8, equal_nan=True)
Dtest_layernorm_stitch.py58 assert np.allclose(e.asnumpy(), o.asnumpy(), rtol, atol, equal_nan=True)
60 assert np.allclose(expect.asnumpy(), output.asnumpy(), rtol, atol, equal_nan=True)
Dtest_dropoutgrad_reducesum_stitch.py61 assert np.allclose(e.asnumpy(), o.asnumpy(), rtol, atol, equal_nan=True)
63 assert np.allclose(expect.asnumpy(), output.asnumpy(), rtol, atol, equal_nan=True)
Dtest_softmax_stitch.py65 assert np.allclose(e.asnumpy(), o.asnumpy(), rtol, atol, equal_nan=True)
67 assert np.allclose(expect.asnumpy(), output.asnumpy(), rtol, atol, equal_nan=True)
Dtest_optimize_assign.py69 …lclose(result_open_gk.asnumpy(), result_close_gk.asnumpy(), rtol=1.e-4, atol=1.e-7, equal_nan=True)
87 …lclose(result_open_gk.asnumpy(), result_close_gk.asnumpy(), rtol=1.e-4, atol=1.e-7, equal_nan=True)
Dtest_gelu.py59 res = np.allclose(expect.asnumpy(), result.asnumpy(), rtol=1.e-4, atol=1.e-4, equal_nan=True)
83 res = np.allclose(expect.asnumpy(), result.asnumpy(), rtol=1.e-4, atol=1.e-4, equal_nan=True)
Dtest_simplify.py85 atol=1.e-7, equal_nan=True)
98 atol=1.e-7, equal_nan=True)
/third_party/mindspore/mindspore/numpy/
Dlogic_ops.py452 def isclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False): argument
494 if not isinstance(equal_nan, bool):
511 if equal_nan:
745 def array_equal(a1, a2, equal_nan=False): argument
778 if not isinstance(equal_nan, bool):
782 if equal_nan:
/third_party/mindspore/tests/st/mix_precision/
Dutils.py37 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
39 assert np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan)
40 elif not np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan):
/third_party/mindspore/tests/st/auto_monad/
Dtest_auto_monad_expression.py51 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
53 assert np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan)
54 elif not np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan):
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_mixed_precision_cells.py80 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
82 assert np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan)
83 elif not np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan):
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py85 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
87 assert np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan)
88 elif not np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan):
/third_party/mindspore/tests/st/auto_parallel/
Doptimizer_parallel.py55 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
57 assert np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan)
58 elif not np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan):
Dmultifieldembeddinglookup_parallel.py51 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
53 assert np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan)
54 elif not np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan):
Dparallel_strategy_search.py57 def allclose_nparray(data_expected, data_me, rtol, atol, equal_nan=True): argument
59 assert np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan)
60 elif not np.allclose(data_expected, data_me, rtol, atol, equal_nan=equal_nan):

12