onnx/common/constants.h

0001 // Copyright (c) ONNX Project Contributors
0002 //
0003 // SPDX-License-Identifier: Apache-2.0
0004
0005 #pragma once
0006
0007 #include <string>
0008
0009 namespace ONNX_NAMESPACE {
0010 // For ONNX op/function registration.
0011
0012 // ONNX domains.
0013 constexpr const char* AI_ONNX_ML_DOMAIN = "ai.onnx.ml";
0014 constexpr const char* AI_ONNX_TRAINING_DOMAIN = "ai.onnx.training";
0015 constexpr const char* AI_ONNX_PREVIEW_TRAINING_DOMAIN = "ai.onnx.preview.training";
0016 // The following two are equivalent in an onnx proto representation.
0017 constexpr const char* ONNX_DOMAIN = "";
0018 constexpr const char* AI_ONNX_DOMAIN = "ai.onnx";
0019
0020 inline std::string NormalizeDomain(const std::string& domain) {
0021   return (domain == AI_ONNX_DOMAIN) ? ONNX_DOMAIN : domain;
0022 }
0023
0024 inline bool IsOnnxDomain(const std::string& domain) {
0025   return (domain == AI_ONNX_DOMAIN) || ((domain == ONNX_DOMAIN));
0026 }
0027
0028 constexpr bool OPTIONAL_VALUE = false;
0029
0030 // For dimension denotation.
0031 constexpr const char* DATA_BATCH = "DATA_BATCH";
0032 constexpr const char* DATA_CHANNEL = "DATA_CHANNEL";
0033 constexpr const char* DATA_TIME = "DATA_TIME";
0034 constexpr const char* DATA_FEATURE = "DATA_FEATURE";
0035 constexpr const char* FILTER_IN_CHANNEL = "FILTER_IN_CHANNEL";
0036 constexpr const char* FILTER_OUT_CHANNEL = "FILTER_OUT_CHANNEL";
0037 constexpr const char* FILTER_SPATIAL = "FILTER_SPATIAL";
0038
0039 // For type denotation.
0040 constexpr const char* TENSOR = "TENSOR";
0041 constexpr const char* IMAGE = "IMAGE";
0042 constexpr const char* AUDIO = "AUDIO";
0043 constexpr const char* TEXT = "TEXT";
0044
0045 } // namespace ONNX_NAMESPACE