Home
last modified time | relevance | path

Searched refs:judge_result_correct (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/st/ops/cpu/
Dtest_sparse_tensor_dense_matmul_op.py46 def judge_result_correct(result, expect): function
68 judge_result_correct(out_ms.asnumpy(), out_np)
73 judge_result_correct(grad_ms[1].asnumpy(), expect_values_grad)
78 judge_result_correct(grad_ms[2].asnumpy(), expect_dense_grad)
98 judge_result_correct(out_ms.asnumpy(), out_np)
103 judge_result_correct(grad_ms[1].asnumpy(), expect_values_grad)
108 judge_result_correct(grad_ms[2].asnumpy(), expect_dense_grad)
128 judge_result_correct(out_ms.asnumpy(), out_np)
133 judge_result_correct(grad_ms[1].asnumpy(), expect_values_grad)
138 judge_result_correct(grad_ms[2].asnumpy(), expect_dense_grad)
[all …]
Dtest_sparse_to_dense_op.py46 def judge_result_correct(result, expect): function
61 judge_result_correct(forward_output.asnumpy(), expect_forward_output)
67 judge_result_correct(grad_output[1].asnumpy(), expect_grad_output)
88 judge_result_correct(forward_output.asnumpy(), expect_forward_output)
94 judge_result_correct(grad_output[1].asnumpy(), expect_grad_output)
104 judge_result_correct(forward_output.asnumpy(), expect_forward_output)
110 judge_result_correct(grad_output[1].asnumpy(), expect_grad_output)
Dtest_matmul.py34 def judge_result_correct(result, expect): function
50 judge_result_correct(output.asnumpy(), expect)
67 judge_result_correct(output.asnumpy(), expect)
82 judge_result_correct(output.asnumpy(), expect)
97 judge_result_correct(output.asnumpy(), expect)
115 judge_result_correct(output.asnumpy(), expect)
Dtest_batch_matmul.py35 def judge_result_correct(result, expect): function
59 judge_result_correct(output.asnumpy(), expect)
84 judge_result_correct(output.asnumpy(), expect)
109 judge_result_correct(output.asnumpy(), expect)
134 judge_result_correct(output.asnumpy(), expect)
159 judge_result_correct(output.asnumpy(), expect)
/third_party/mindspore/tests/st/ops/gpu/
Dtest_hswish_op.py53 def judge_result_correct(result, expect): function
65 judge_result_correct(output.asnumpy(), expect)
71 judge_result_correct(output[0].asnumpy(), expect)
Dtest_prelu_op.py46 def judge_result_correct(result, expect): function
58 judge_result_correct(forward_output, expect_forward)
62 judge_result_correct(backward_output[0], expect_dx)
63 judge_result_correct(backward_output[1], expect_dw)