conv_impl.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_impl {
    STD,       
    VEC,       
    CUDNN,     
    FFT_STD,   
    FFT_MKL,   
    FFT_CUFFT, 
    EGBLAS     
};

enum class conv4_impl {
    STD,       
    VEC,       
    CUDNN,     
    FFT_STD,   
    FFT_MKL,   
    FFT_CUFFT, 
    BLAS_VEC,  
    BLAS_MKL   
};

enum class conv_multi_impl {
    STD,           
    VEC,           
    VALID_FFT_MKL, 
    FFT_STD,       
    FFT_MKL,       
    FFT_CUFFT,     
    BLAS_VEC,      
    BLAS_MKL,      
    CUDNN          
};

} //end of namespace etl