ML_Types.h

Go to the documentation of this file.

/*
 * PROPRIETARY INFORMATION.  This software is proprietary to
 * Side Effects Software Inc., and is not to be reproduced,
 * transmitted, or disclosed in any way without written permission.
 *
 * COMMENTS:
 *
 */

#pragma once

#include "ML_API.h"

/*
 * Enumeration of supported ONNX execution providers
 */
enum class ML_ExecutionProvider
{
    /// Automatically determine the best provider based on the platform
    /// and hardware available
    AUTOMATIC,

    /// CPU inference provider, works on all platforms
    CPU,

    /// CUDA/cudNN inference provider, works on Windows and Linux assuming
    /// the GPU driver is new enough and cudNN is installed on the system
    CUDA,

    /// Uses DirectML, only supported on Windows
    DIRECTML,

    /// Uses CoreML, only supported on macOS
    COREML,

    /// The total number of providers
    COUNT
};