| 394 | } |
| 395 | |
| 396 | void CheckerHelper::check_tensors( |
| 397 | const TensorValueArray& expected, const TensorValueArray& computed) { |
| 398 | for (size_t i = 0; i < expected.size(); ++i) { |
| 399 | if (expected[i].layout.ndim == 0 || m_bypass.find(i) != m_bypass.end()) |
| 400 | continue; |
| 401 | if (m_allow_invalid_check) { |
| 402 | MEGDNN_ASSERT_TENSOR_EQ_EPS_AVG_ALLOW_INVALID( |
| 403 | expected[i], computed[i], m_epsilon, m_max_avg_error, |
| 404 | m_max_avg_biased_error); |
| 405 | } else { |
| 406 | MEGDNN_ASSERT_TENSOR_EQ_EPS_AVG( |
| 407 | expected[i], computed[i], m_epsilon, m_max_avg_error, |
| 408 | m_max_avg_biased_error); |
| 409 | } |
| 410 | } |
| 411 | } |
| 412 | |
| 413 | void CheckerHelper::copy_tensors_to_device( |
| 414 | const TensorValueArray& dest, const TensorValueArray& src) { |