mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-30 23:18:20 +00:00
00b1e79e04
**Description**: This PR intends to enable WebNN EP in ONNX Runtime Web. It translates the ONNX nodes by [WebNN API](https://webmachinelearning.github.io/webnn/), which is implemented in C++ and uses Emscripten [Embind API](https://emscripten.org/docs/porting/connecting_cpp_and_javascript/embind.html#). Temporarily using preferred layout **NHWC** for WebNN graph partitions since the restriction in WebNN XNNPack backend implementation and the ongoing [discussion](https://github.com/webmachinelearning/webnn/issues/324) in WebNN spec that whether WebNN should support both 'NHWC' and 'NCHW' layouts. No WebNN native EP, only for Web. **Motivation and Context**: Allow ONNXRuntime Web developers to access WebNN API to benefit from hardware acceleration. **WebNN API Implementation Status in Chromium**: - Tracked in Chromium issue: [#1273291](https://bugs.chromium.org/p/chromium/issues/detail?id=1273291) - **CPU device**: based on XNNPack backend, and had been available on Chrome Canary M112 behind "#enable-experimental-web-platform-features" flag for Windows and Linux platforms. Further implementation for more ops is ongoing. - **GPU device**: based on DML, implementation is ongoing. **Open**: - GitHub CI: WebNN currently is only available on Chrome Canary/Dev with XNNPack backend for Linux and Windows. This is an open to reviewers to help identify which GitHub CI should involved the WebNN EP and guide me to enable it. Thanks!
58 lines
3.1 KiB
C++
58 lines
3.1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include <stddef.h> // needed for size_t on some platforms
|
|
|
|
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 = "";
|
|
// NOTE: Node::Init converts kOnnxDomainAlias to kOnnxDomain, so all Node instances use kOnnxDomain.
|
|
constexpr const char* kOnnxDomainAlias = "ai.onnx";
|
|
constexpr const char* kMLDomain = "ai.onnx.ml";
|
|
constexpr const char* kMSDomain = "com.microsoft";
|
|
constexpr const char* kPytorchAtenDomain = "org.pytorch.aten";
|
|
constexpr const char* kMSExperimentalDomain = "com.microsoft.experimental";
|
|
constexpr const char* kMSNchwcDomain = "com.microsoft.nchwc";
|
|
constexpr const char* kMSInternalNHWCDomain = "com.ms.internal.nhwc";
|
|
constexpr const char* kMSDmlDomain = "com.microsoft.dml";
|
|
constexpr const char* kNGraphDomain = "com.intel.ai";
|
|
constexpr const char* kMIGraphXDomain = "";
|
|
constexpr const char* kVitisAIDomain = "com.xilinx";
|
|
|
|
// This is moved from the OrtApis::GetAvailableProviders implementation
|
|
// where it is enforced
|
|
constexpr size_t kMaxExecutionProviderNameLen = 30;
|
|
|
|
constexpr const char* kCpuExecutionProvider = "CPUExecutionProvider";
|
|
constexpr const char* kCudaExecutionProvider = "CUDAExecutionProvider";
|
|
constexpr const char* kDnnlExecutionProvider = "DnnlExecutionProvider";
|
|
constexpr const char* kOpenVINOExecutionProvider = "OpenVINOExecutionProvider";
|
|
constexpr const char* kVitisAIExecutionProvider = "VitisAIExecutionProvider";
|
|
constexpr const char* kTensorrtExecutionProvider = "TensorrtExecutionProvider";
|
|
constexpr const char* kNnapiExecutionProvider = "NnapiExecutionProvider";
|
|
constexpr const char* kQnnExecutionProvider = "QNNExecutionProvider";
|
|
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";
|
|
constexpr const char* kRocmExecutionProvider = "ROCMExecutionProvider";
|
|
constexpr const char* kCoreMLExecutionProvider = "CoreMLExecutionProvider";
|
|
constexpr const char* kJsExecutionProvider = "JsExecutionProvider";
|
|
constexpr const char* kSnpeExecutionProvider = "SNPEExecutionProvider";
|
|
constexpr const char* kTvmExecutionProvider = "TvmExecutionProvider";
|
|
constexpr const char* kXnnpackExecutionProvider = "XnnpackExecutionProvider";
|
|
constexpr const char* kWebNNExecutionProvider = "WebNNExecutionProvider";
|
|
constexpr const char* kCannExecutionProvider = "CANNExecutionProvider";
|
|
constexpr const char* kAzureExecutionProvider = "AzureExecutionProvider";
|
|
|
|
constexpr const char* kExecutionProviderSharedLibraryPath = "shared_lib_path";
|
|
constexpr const char* kExecutionProviderSharedLibraryEntry = "provider_factory_entry_point";
|
|
|
|
} // namespace onnxruntime
|