1 | /* |
2 | * SPDX-License-Identifier: Apache-2.0 |
3 | */ |
4 | |
5 | #pragma once |
6 | |
7 | #include <string> |
8 | |
9 | namespace ONNX_NAMESPACE { |
10 | // For ONNX op/function registration. |
11 | |
12 | // ONNX domains. |
13 | constexpr const char* AI_ONNX_ML_DOMAIN = "ai.onnx.ml" ; |
14 | constexpr const char* AI_ONNX_TRAINING_DOMAIN = "ai.onnx.training" ; |
15 | constexpr const char* AI_ONNX_PREVIEW_TRAINING_DOMAIN = "ai.onnx.preview.training" ; |
16 | constexpr const char* ONNX_DOMAIN = "" ; |
17 | constexpr bool OPTIONAL_VALUE = false; |
18 | |
19 | // For dimension denotation. |
20 | constexpr const char* DATA_BATCH = "DATA_BATCH" ; |
21 | constexpr const char* DATA_CHANNEL = "DATA_CHANNEL" ; |
22 | constexpr const char* DATA_TIME = "DATA_TIME" ; |
23 | constexpr const char* DATA_FEATURE = "DATA_FEATURE" ; |
24 | constexpr const char* FILTER_IN_CHANNEL = "FILTER_IN_CHANNEL" ; |
25 | constexpr const char* FILTER_OUT_CHANNEL = "FILTER_OUT_CHANNEL" ; |
26 | constexpr const char* FILTER_SPATIAL = "FILTER_SPATIAL" ; |
27 | |
28 | // For type denotation. |
29 | constexpr const char* TENSOR = "TENSOR" ; |
30 | constexpr const char* IMAGE = "IMAGE" ; |
31 | constexpr const char* AUDIO = "AUDIO" ; |
32 | constexpr const char* TEXT = "TEXT" ; |
33 | |
34 | } // namespace ONNX_NAMESPACE |
35 | |