Method get_workspace_in_bytes

dnn/src/fallback/warp_perspective/opr_impl.cpp:43–52

namespace fallback {

size_t WarpPerspectiveImpl::get_workspace_in_bytes(
        const TensorLayout&, const TensorLayout&, const TensorLayout&,
        const TensorLayout& dst) {
    if (param().format == param::WarpPerspective::Format::NCHW) {
        size_t OH = dst.shape[2], OW = dst.shape[3];
        return get_bundle(OH, OW).total_size_in_bytes();
    } else {
        return 0;
    }
}

size_t WarpPerspectiveImpl::get_workspace_in_bytes(
        const TensorLayoutArray&, const TensorLayout&, const TensorLayout&,