mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-26 03:00:54 +00:00
* Add ArmNN Execution Provider Add a new execution provider targeting Arm architecture based on ArmNN. Validated on NXP i.MX8QM CPU with ResNet50, MobileNetv2 and VGG models. reviewed-by: mike.caraman@nxp.com * Minor fixes - renamed onnxruntime_ARMNN_RELU_USECPU to onnxruntime_ARMNN_RELU_USE_CPU - fixed acl typo * remove extra includes. added exception for ArmNN in test * fix indentation * Separated the activation implementation from the cpu and fixed the blockage from the endif Co-authored-by: Andrei-Alexandru <andrei-alexandru.avram@nxp.com>
41 lines
2 KiB
C++
41 lines
2 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <unordered_set>
|
|
#include <vector>
|
|
|
|
#include "core/common/common.h"
|
|
|
|
namespace onnxruntime {
|
|
constexpr const char* kNoOp = "NoOp";
|
|
constexpr const char* kConstant = "Constant";
|
|
constexpr const char* kFunctionOp = "_kFunctionOp";
|
|
constexpr const char* kConstantValue = "value";
|
|
constexpr const char* kOnnxDomain = "";
|
|
constexpr const char* kOnnxDomainAlias = "ai.onnx";
|
|
constexpr const char* kMLDomain = "ai.onnx.ml";
|
|
constexpr const char* kMSDomain = "com.microsoft";
|
|
constexpr const char* kMSNchwcDomain = "com.microsoft.nchwc";
|
|
constexpr const char* kMSFeaturizersDomain = "com.microsoft.mlfeaturizers";
|
|
constexpr const char* kMSDmlDomain = "com.microsoft.dml";
|
|
constexpr const char* kNGraphDomain = "com.intel.ai";
|
|
constexpr const char* kMIGraphXDomain = "";
|
|
constexpr const char* kVitisAIDomain = "com.xilinx";
|
|
constexpr const char* kCpuExecutionProvider = "CPUExecutionProvider";
|
|
constexpr const char* kCudaExecutionProvider = "CUDAExecutionProvider";
|
|
constexpr const char* kDnnlExecutionProvider = "DnnlExecutionProvider";
|
|
constexpr const char* kNGraphExecutionProvider = "NGRAPHExecutionProvider";
|
|
constexpr const char* kOpenVINOExecutionProvider = "OpenVINOExecutionProvider";
|
|
constexpr const char* kNupharExecutionProvider = "NupharExecutionProvider";
|
|
constexpr const char* kVitisAIExecutionProvider = "VitisAIExecutionProvider";
|
|
constexpr const char* kTensorrtExecutionProvider = "TensorrtExecutionProvider";
|
|
constexpr const char* kNnapiExecutionProvider = "NnapiExecutionProvider";
|
|
constexpr const char* kRknpuExecutionProvider = "RknpuExecutionProvider";
|
|
constexpr const char* kDmlExecutionProvider = "DmlExecutionProvider";
|
|
constexpr const char* kMIGraphXExecutionProvider = "MIGraphXExecutionProvider";
|
|
constexpr const char* kAclExecutionProvider = "ACLExecutionProvider";
|
|
constexpr const char* kArmNNExecutionProvider = "ArmNNExecutionProvider";
|
|
} // namespace onnxruntime
|