| 28 | } |
| 29 | |
| 30 | void LSQForward::check_exec( |
| 31 | const TensorLayout& input, const TensorLayout& scale, |
| 32 | const TensorLayout& zero_point, const TensorLayout& grad_scale, |
| 33 | const TensorLayout& output, size_t workspace_in_bytes) { |
| 34 | check_layout_fwd(input, scale, zero_point, grad_scale, output); |
| 35 | auto required_workspace_space = |
| 36 | get_workspace_in_bytes(input, scale, zero_point, grad_scale, output); |
| 37 | megdnn_assert(workspace_in_bytes >= required_workspace_space); |
| 38 | } |
| 39 | |
| 40 | void LSQBackward::check_exec( |
| 41 | const TensorLayout& diff, const TensorLayout& input, const TensorLayout& scale, |
nothing calls this directly
no test coverage detected