conv_select.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c) 2014-2023 Baptiste Wicht
// Distributed under the terms of the MIT License.
// (See accompanying file LICENSE or copy at
//  http://opensource.org/licenses/MIT)
//=======================================================================

#pragma once

namespace etl {

enum class conv_type {
    VALID,       
    VALID_MULTI, 
    SAME,        
    SAME_MULTI,  
    FULL,        
    FULL_MULTI   
};

namespace detail {

template <typename I, typename K, typename C>
inline bool select_parallel(const I& /*input*/, const K& kernel, C&& conv) {
    if ((is_parallel && !local_context().serial) || (parallel_support && local_context().parallel)) {
        return etl::size(conv) >= conv1_parallel_threshold_conv && etl::size(kernel) >= conv1_parallel_threshold_kernel;
    } else {
        return false;
    }
}

} //end of namespace detail

} //end of namespace etl

#include "conv_normal_select.hpp"
#include "conv_multi_select.hpp"
#include "conv_4d_select.hpp"