Method get_workspace_bundle

dnn/src/cuda/group_norm/opr_impl.cpp:16–19

}

WorkspaceBundle GroupNormForwardImpl::get_workspace_bundle(
        size_t N, size_t G, size_t dtype_size, void* raw_ptr) {
    return {raw_ptr, {N * G * dtype_size}, handle()->alignment_requirement()};
}

void GroupNormForwardImpl::exec(
        _megdnn_tensor_in data, _megdnn_tensor_in weight, _megdnn_tensor_in bias,