| 91 | } |
| 92 | |
| 93 | void GeneralNormForward::check_exec( |
| 94 | const TensorLayout& data, const TensorLayout& weight, const TensorLayout& bias, |
| 95 | const TensorLayout& dst, const TensorLayout& mean, const TensorLayout& rstd, |
| 96 | size_t workspace_in_bytes) { |
| 97 | check_layout_fwd(data, weight, bias, dst, mean, rstd); |
| 98 | size_t required_workspace_in_bytes = |
| 99 | get_workspace_in_bytes(data, weight, bias, dst, mean, rstd); |
| 100 | megdnn_assert(workspace_in_bytes >= required_workspace_in_bytes); |
| 101 | } |
| 102 | |
| 103 | void GeneralNormBackward::deduce_layout( |
| 104 | const TensorLayout& diff, const TensorLayout& data, const TensorLayout& weight, |
nothing calls this directly
no test coverage detected