Method get_workspace_bundle

dnn/src/cambricon/group_norm/opr_impl.cpp:20–33

}

WorkspaceBundle GroupNormForwardImpl::get_workspace_bundle(
        const TensorLayout& data, void* raw_ptr) {
    auto handle = concrete_handle(this->handle());
    int group_num = static_cast<int>(param().group);
    size_t channel = data.shape[1];
    size_t workspace_size = 0;
    CnnlTensorDescriptor x_desc;
    x_desc.set(data, CNNL_LAYOUT_NCHW);
    cnnl_check(cnnlGetGroupNormForwardWorkspaceSize(
            handle->cnnl_handle(), group_num, x_desc.desc(), &workspace_size));
    SmallVector<size_t> sizes_in_bytes;
    sizes_in_bytes.push_back(workspace_size);
    return {raw_ptr, sizes_in_bytes, handle->alignment_requirement()};
}

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