| 13 | } |
| 14 | |
| 15 | void LRNForward::check_exec( |
| 16 | const TensorLayout& src, const TensorLayout& dst, size_t workspace_in_bytes) { |
| 17 | check_param(); |
| 18 | megdnn_assert_contiguous(src); |
| 19 | megdnn_assert_eq_layout(src, dst); |
| 20 | |
| 21 | megdnn_assert(src.dtype.category() == DTypeCategory::FLOAT); |
| 22 | auto required_workspace_in_bytes = get_workspace_in_bytes(src, dst); |
| 23 | megdnn_assert(workspace_in_bytes >= required_workspace_in_bytes); |
| 24 | } |
| 25 | |
| 26 | void LRNBackward::check_exec( |
| 27 | const TensorLayout& src, const TensorLayout& dst, const TensorLayout& diff, |
nothing calls this directly
no test coverage detected