| 144 | namespace naive { |
| 145 | |
| 146 | size_t GroupNormBackwardImpl::get_workspace_in_bytes( |
| 147 | const TensorLayout&, const TensorLayout& data, const TensorLayout&, |
| 148 | const TensorLayout&, const TensorLayout& rstd, const TensorLayout&, |
| 149 | const TensorLayout&, const TensorLayout&) { |
| 150 | size_t N = data.shape[0]; |
| 151 | size_t C = data.shape[1]; |
| 152 | size_t G = rstd.shape[1]; |
| 153 | return get_workspace_bundle(N, C, G, data.dtype.size()).total_size_in_bytes(); |
| 154 | } |
| 155 | |
| 156 | WorkspaceBundle GroupNormBackwardImpl::get_workspace_bundle( |
| 157 | size_t N, size_t C, size_t G, size_t dtype_size, void* raw_ptr) { |
nothing calls this directly
no test coverage detected