Function get_workspace_bundle

dnn/src/cambricon/convolution/forward/algo.cpp:49–58

}

WorkspaceBundle get_workspace_bundle(
        const std::pair<TensorLayoutArray, std::unique_ptr<ConvBiasForward>>&
                subopr_config,
        const ConvolutionForwardImpl::AlgoBase::SizeArgs& args) {
    size_t bias_size = subopr_config.first[0].access_bytes();
    size_t conv_work_size = subopr_config.second->get_workspace_in_bytes(
            *args.layout_src, *args.layout_filter, subopr_config.first[0],
            subopr_config.first[1], *args.layout_dst, nullptr);
    return {nullptr, {bias_size, conv_work_size}};
}

}  // namespace