Method get_workspace_in_bytes

dnn/src/cuda/group_norm/opr_impl.cpp:8–14

namespace cuda {

size_t GroupNormForwardImpl::get_workspace_in_bytes(
        const TensorLayout&, const TensorLayout&, const TensorLayout&,
        const TensorLayout&, const TensorLayout&, const TensorLayout& rstd) {
    size_t N = rstd.shape[0];
    size_t G = rstd.shape[1];
    return get_workspace_bundle(N, G, rstd.dtype.size()).total_size_in_bytes();
}

WorkspaceBundle GroupNormForwardImpl::get_workspace_bundle(
        size_t N, size_t G, size_t dtype_size, void* raw_ptr) {