|
|
|
/**
|
|
|
|
* Copyright 2020 Huawei Technologies Co., Ltd
|
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "framework/common/types.h"
|
|
|
|
#include "graph/types.h"
|
|
|
|
|
|
|
|
namespace ge {
|
|
|
|
// dump
|
|
|
|
const std::string DUMP_MODEL = "model_name";
|
|
|
|
const std::string DUMP_ALL_MODEL = "ALL_MODEL_NEED_DUMP_AND_IT_IS_NOT_A_MODEL_NAME";
|
|
|
|
const std::string DUMP_STATUS = "status";
|
|
|
|
const std::string DUMP_LAYER = "layer";
|
|
|
|
const std::string DUMP_FILE_PATH = "path";
|
|
|
|
const std::string DUMP_MODE = "dump_mode";
|
|
|
|
|
|
|
|
// op debug mode
|
|
|
|
const std::string OP_DEBUG_AICORE = "aicore_overflow";
|
|
|
|
const std::string OP_DEBUG_ATOMIC = "atomic_overflow";
|
|
|
|
const std::string OP_DEBUG_ALL = "all";
|
|
|
|
|
|
|
|
const int DEFAULT_FORMAT = static_cast<const int>(ge::FORMAT_NCHW);
|
|
|
|
// Supported public property names
|
|
|
|
const std::string PROP_OME_START_TIME = "ome_start_time"; // start time
|
|
|
|
const std::string PROP_OME_DUMP_PATH = "ome_dump_path"; // dump path
|
|
|
|
const std::string PROP_OME_LOG_PATH = "ome_log_path"; // log path
|
|
|
|
|
|
|
|
// Profile related constant
|
|
|
|
const uint32_t CCE_PROFILE_ON = 0;
|
|
|
|
const uint32_t CCE_PROFILE_OFF = 1;
|
|
|
|
const std::string OME_PROFILE = "OME_PROFILE";
|
|
|
|
const std::string CCE_PROFILE = "CCE_PROFILE";
|
|
|
|
const std::string RTS_PROFILE = "RTS_PROFILE";
|
|
|
|
const std::string PROFILER_JOBCTX = "profiler_jobctx";
|
|
|
|
const std::string PROFILER_TARGET_PATH = "profiler_target_path";
|
|
|
|
const std::string RTS_PROFILE_PATH = "RTS_PATH";
|
|
|
|
const std::string PROFILE_STOP_KEY = "stop";
|
|
|
|
const std::string PROFILE_STOP_VALUE = "enable";
|
|
|
|
const std::map<std::string, std::string> PROFILE_COMPONENT_MAP{
|
|
|
|
{"ome", OME_PROFILE},
|
|
|
|
{"cce", CCE_PROFILE},
|
|
|
|
{"runtime", RTS_PROFILE},
|
|
|
|
};
|
|
|
|
const std::string PROFILE_CONFIG = "config";
|
|
|
|
const std::string PROFILE_MODEL_ID = "modelId";
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(DATA, "Data");
|
|
|
|
REGISTER_OPTYPE_DEFINE(AIPPDATA, "AippData");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONVOLUTION, "Convolution");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CORRELATION, "Correlation");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CORRELATIONV2, "Correlation_V2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DECONVOLUTION, "Deconvolution");
|
|
|
|
REGISTER_OPTYPE_DEFINE(POOLING, "Pooling");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ELTWISE, "Eltwise");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RELU, "ReLU");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RELU6, "ReLU6");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SIGMOID, "Sigmoid");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ABSVAL, "AbsVal");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TANH, "TanH");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PRELU, "PReLU");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BATCHNORM, "BatchNorm");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FUSIONBATCHNORM, "FusionBatchNorm");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SCALE, "Scale");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FULL_CONNECTION, "FullConnection");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SOFTMAX, "Softmax");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PLUS, "Plus");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ACTIVATION, "Activation");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FLATTEN, "Flatten");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ADD, "Add");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SUB, "Sub");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MUL, "Mul");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MATMUL, "MatMul");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RSQRT, "Rsqrt");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BIASADD, "BiasAdd");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RESHAPE, "Reshape");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFORMAT, "ReFormat");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEPCONVOLUTION, "ConvolutionDepthwise");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DROPOUT, "Dropout");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DROPOUTGENMASK, "DropOutGenMask");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DROPOUTDOMASK, "DropOutDoMask");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DROPOUTDOMASKV3, "DropOutDoMaskV3");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DROPOUTDOMASKV3D, "DropOutDoMaskV3D");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONCAT, "Concat");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ROIPOOLING, "ROIPooling");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PROPOSAL, "Proposal");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FSRDETECTIONOUTPUT, "FSRDetectionOutput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DETECTIONPOSTPROCESS, "Detectpostprocess");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LRN, "LRN");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TRANSDATA, "TransData");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PERMUTE, "Permute");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDNORMALIZE, "SSDNormalize");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDPRIORBOX, "SSDPriorBox");
|
|
|
|
REGISTER_OPTYPE_DEFINE(NETOUTPUT, "NetOutput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDDETECTIONOUTPUT, "SSDDetectionOutput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFINEDETDETECTIONOUTPUT, "RefinedetDetectionOutput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CHANNELAXPY, "ChannelAxpy");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PSROIPOOLING, "PSROIPooling");
|
|
|
|
REGISTER_OPTYPE_DEFINE(POWER, "Power");
|
|
|
|
REGISTER_OPTYPE_DEFINE(POW, "Pow");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ROIALIGN, "ROIAlign");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PYTHON, "Python");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FREESPACEEXTRACT, "FreespaceExtract");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPATIALTF, "SpatialTransform");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SHAPE, "Shape");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SHAPEN, "ShapeN");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ARGMAX, "ArgMax");
|
|
|
|
REGISTER_OPTYPE_DEFINE(GATHERND, "GatherNd");
|
|
|
|
REGISTER_OPTYPE_DEFINE(GATHER, "Gather");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REALDIV, "RealDiv");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PACK, "Pack");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SLICE, "Slice");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SLICED, "SliceD");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FLOORDIV, "FloorDiv");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SQUEEZE, "Squeeze");
|
|
|
|
REGISTER_OPTYPE_DEFINE(UNSQUEEZE, "Unsqueeze");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STRIDEDSLICE, "StridedSlice");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RANGE, "Range");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RPNPROPOSALS, "RpnProposals");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DECODEBBOX, "DecodeBbox");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PAD, "Pad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PADV2, "PadV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MIRRORPAD, "MirrorPad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TILE, "Tile");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SIZE, "Size");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CLIPBOXES, "ClipBoxes");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTRCNNPREDICTIONS, "FastrcnnPredictions");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPLIT, "Split");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPLITV, "SplitV");
|
|
|
|
REGISTER_OPTYPE_DEFINE(EXPANDDIMS, "ExpandDims");
|
|
|
|
REGISTER_OPTYPE_DEFINE(EMPTY, "Empty");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MEAN, "Mean");
|
|
|
|
REGISTER_OPTYPE_DEFINE(GREATER, "Greater");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SWITCH, "Switch");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SWITCHN, "SwitchN");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MERGE, "Merge");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SYMBOLICGRADIENT, "SymbolicGradient");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REMOTECALL, "RemoteCall");
|
|
|
|
REGISTER_OPTYPE_DEFINE(_IF, "_If");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STATELESSIF, "StatelessIf");
|
|
|
|
REGISTER_OPTYPE_DEFINE(IF, "If");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CASE, "Case");
|
|
|
|
REGISTER_OPTYPE_DEFINE(_WHILE, "_While");
|
|
|
|
REGISTER_OPTYPE_DEFINE(WHILE, "While");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STATELESSWHILE, "StatelessWhile");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FOR, "For");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PARTITIONEDCALL, "PartitionedCall");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STATEFULPARTITIONEDCALL, "StatefulPartitionedCall");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FAKEPARAM, "FakeParam");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TRANSPOSE, "Transpose");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TRANSPOSED, "TransposeD");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CAST, "Cast");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REGION, "Region");
|
|
|
|
REGISTER_OPTYPE_DEFINE(YOLO, "Yolo");
|
|
|
|
REGISTER_OPTYPE_DEFINE(YOLODETECTIONOUTPUT, "YoloDetectionOutput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FILL, "Fill");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REVERSE, "Reverse");
|
|
|
|
REGISTER_OPTYPE_DEFINE(UNPACK, "Unpack");
|
|
|
|
REGISTER_OPTYPE_DEFINE(YOLO2REORG, "Yolo2Reorg");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REDUCESUM, "ReduceSum");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SUM, "Sum");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONSTANT, "Const");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RESIZEBILINEAR, "ResizeBilinear");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RESIZEBILINEARGRAD, "ResizeBilinearGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MAXIMUM, "Maximum");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FRAMEWORKOP, "FrameworkOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ARG, "_Arg");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FUSEDBATCHNORMGRAD, "FusedBatchNormGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LSTM, "LSTM");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HIGHWAY, "HighWay");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RNN, "RNN");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ATTENTIONDECODER, "AttentionDecoder");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LOGICAL_NOT, "LogicalNot");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LOGICAL_AND, "LogicalAnd");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LOGICAL_OR, "LogicalOr");
|
|
|
|
REGISTER_OPTYPE_DEFINE(EQUAL, "Equal");
|
|
|
|
REGISTER_OPTYPE_DEFINE(NOTEQUAL, "NotEqual");
|
|
|
|
REGISTER_OPTYPE_DEFINE(INTERP, "Interp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SHUFFLECHANNEL, "ShuffleChannel");
|
|
|
|
REGISTER_OPTYPE_DEFINE(AIPP, "Aipp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MULTISHAPE, "MultiShape");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RECIPROCAL, "Reciprocal");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SELU, "Selu");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ELU, "Elu");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ACOSH, "Acosh");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASINH, "Asinh");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MINIMUM, "Minimum");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CLIP, "Clip");
|
|
|
|
REGISTER_OPTYPE_DEFINE(L2NORMALIZE, "L2Normalize");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CROPANDRESIZE, "CropAndResize");
|
|
|
|
REGISTER_OPTYPE_DEFINE(UNUSEDCONST, "UnusedConst");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPARSETODENSE, "SparseToDense");
|
|
|
|
REGISTER_OPTYPE_DEFINE(NONMAXSUPPRESSION, "NonMaxSuppression");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TOPKV2, "TopKV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(INVERTPERMUTATION, "InvertPermutation");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MULTINOMIAL, "Multinomial");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REVERSESEQUENCE, "ReverseSequence");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REDUCEPROD, "ReduceProd");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REDUCEMAX, "ReduceMax");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REDUCEMIN, "ReduceMin");
|
|
|
|
REGISTER_OPTYPE_DEFINE(EXTRACTIMAGEPATCHES, "ExtractImagePatches");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SQRT, "Sqrt");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REDUCEALL, "ReduceAll");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RESIZENEARESTNEIGHBOR, "ResizeNearestNeighbor");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPACETOBATCHND, "SpaceToBatchND");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BATCHTOSPACEND, "BatchToSpaceND");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASSERT, "Assert");
|
|
|
|
REGISTER_OPTYPE_DEFINE(GREATEREQUAL, "GreaterEqual");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FLOOR, "Floor");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RANDOMUNIFORM, "RandomUniform");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BATCHMATMUL, "BatchMatMul");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPACETODEPTH, "SpaceToDepth");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEPTHTOSPACE, "DepthToSpace");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RINT, "Rint");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ATAN, "Atan");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ATAN2, "Atan2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ATANH, "Atanh");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ACOS, "Acos");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASIN, "Asin");
|
|
|
|
REGISTER_OPTYPE_DEFINE(NEG, "Neg");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LOG, "Log");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TAN, "Tan");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ROUND, "Round");
|
|
|
|
REGISTER_OPTYPE_DEFINE(UPSAMPLE, "Upsample");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FLOORMOD, "FloorMod");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LESS, "Less");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LESSEQUAL, "LessEqual");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ONEHOT, "OneHot");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFSWITCH, "RefSwitch");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFMERGE, "RefMerge");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ENTER, "Enter");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFENTER, "RefEnter");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LOOPCOND, "LoopCond");
|
|
|
|
REGISTER_OPTYPE_DEFINE(NEXTITERATION, "NextIteration");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFNEXTITERATION, "RefNextIteration");
|
|
|
|
REGISTER_OPTYPE_DEFINE(EXIT, "Exit");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFEXIT, "RefExit");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONTROLTRIGGER, "ControlTrigger");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ZEROSLIKE, "ZerosLike");
|
|
|
|
REGISTER_OPTYPE_DEFINE(EXP, "Exp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(WHERE, "Where");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FAKEQUANTWITHMINMAXVARS, "FakeQuantWithMinMaxVars");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SOFTPLUS, "Softplus");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SOFTSIGN, "Softsign");
|
|
|
|
REGISTER_OPTYPE_DEFINE(COSH, "Cosh");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SINH, "Sinh");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SQUAREDDIFFERENCE, "SquaredDifference");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REQUIREDSPACETOBATCHPADDINGS, "RequiredSpaceToBatchPaddings"); // for retinanet scope fusion
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDPOSTPROCESSOR, "SSDPostProcessor");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RETINANETBOXES, "RetinanetBoxes");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RETINAMULTIANCHORS, "RetinaMultiAnchor");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RETINANETCLIPPEDBOXES, "RetinanetClippedBoxes");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RETINANETFILTEREDDETECTIONS, "RetinanetFilteredDetections");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RETINANETPOSTPROCESSOR, "RetinanetPostProcessor");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RETINANETANCHORS, "RetinanetAnchors");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTERRCNNMAP, "FasterRCNNMap");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTERRCNNMAP1, "FasterRCNNMap1");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTERRCNNSECONDSTAGEPOSTPROCESSOR, "FasterRCNNSecondStagePostprocessor");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTERRCNNROIINTERPOOLING, "FasterRCNNROIInterPooling");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTERRCNNFIRSTSTAGEPOSTPROCESSOR, "FasterRCNNFirstStagePostprocessor");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTERRCNNGRIDANCHORGENERATOR, "FasterRCNNGridAnchorGenerator");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ROIINTERPOOLING, "ROIInterPooling");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FASTERRCNNCLIPTOWINDOW, "FasterRCNNClipToWindow");
|
|
|
|
REGISTER_OPTYPE_DEFINE(EMBEDLOOKUP, "EmbedLookup");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HASHLOOKUP, "HashLookup");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LSH_PROJ, "LshProject");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SVDF, "SVDF");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDANCHORGENERATOR, "SSDAnchorGenerator");
|
|
|
|
REGISTER_OPTYPE_DEFINE(IDENTITY, "Identity");
|
|
|
|
REGISTER_OPTYPE_DEFINE(IDENTITYN, "IdentityN");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PLACEHOLDERWITHDEFAULT, "PlaceholderWithDefault");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SELECT, "Select");
|
|
|
|
REGISTER_OPTYPE_DEFINE(GETSPAN, "GetSpan");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STOPGRADIENT, "StopGradient");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PREVENTGRADIENT, "PreventGradient");
|
|
|
|
REGISTER_OPTYPE_DEFINE(GUARANTEECONST, "GuaranteeConst");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BROADCASTGRADIENTARGS, "BroadcastGradientArgs");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BROADCASTARGS, "BroadcastArgs");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONFUSIONMATRIX, "ConfusionMatrix");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RANK, "Rank");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PLACEHOLDER, "PlaceHolder");
|
|
|
|
REGISTER_OPTYPE_DEFINE(END, "End");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BASICLSTMCELL, "BasicLSTMCell");
|
|
|
|
REGISTER_OPTYPE_DEFINE(GETNEXT, "GetNext");
|
|
|
|
REGISTER_OPTYPE_DEFINE(INITDATA, "InitData");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REFIDENTITY, "RefIdentity");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BITCAST, "Bitcast");
|
|
|
|
|
|
|
|
/***************Ann special operator*************************/
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_MEAN, "AnnMean");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_CONVOLUTION, "AnnConvolution");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_DEPCONVOLUTION, "AnnDepthConv");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_FULLCONNECTION, "AnnFullConnection");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_NETOUTPUT, "AnnNetOutput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_DATA, "AnnData");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_RESHAPE, "AnnReshape");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_ADD, "AnnAdd");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_MUL, "AnnMul");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_SUB, "AnnSub");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_DIV, "AnnDiv");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_DEQUANTIZE, "AnnDequant");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_QUANTIZE, "AnnQuant");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_PAD, "AnnPad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANN_RESIZE_BILINEAR, "AnnResizeBilinear");
|
|
|
|
|
|
|
|
/***************************************************/
|
|
|
|
/******************Training operator*************************/
|
|
|
|
REGISTER_OPTYPE_DEFINE(GATHERV2, "GatherV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONVGRADFILTER, "Conv2DBackpropFilter");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONV2D, "Conv2D");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONV2DBACKPROPINPUT, "Conv2DBackpropInput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FUSEDBATCHNORM, "FusedBatchNorm");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BIASADDGRAD, "BiasAddGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ACTIVATIONGRAD, "ReluGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MAXPOOLWITHARGMAX, "MaxPoolWithArgmax");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MAXPOOLGRADWITHARGMAX, "MaxPoolGradWithArgmax");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPARSESOFTMAXCROSSENTROPYWITHLOGITS, "SparseSoftmaxCrossEntropyWithLogits");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SNAPSHOT, "Snapshot");
|
|
|
|
REGISTER_OPTYPE_DEFINE(VAR, "Var");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MEANGRAD, "MeanGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TRANSLATE, "Translate");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ADDN, "AddN");
|
|
|
|
REGISTER_OPTYPE_DEFINE(L2LOSS, "L2Loss");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MULTIPLY, "Multiply");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HUBERLOSSGRAD, "HuberLossGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HUBERLOSS, "HuberLoss");
|
|
|
|
REGISTER_OPTYPE_DEFINE(NEGATIVE, "Negative");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDCAST, "SSDCast");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPARSESOFTMAXCROSSENTROPY, "SsdSparseSoftmaxCrossEntropy");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SPARSESOFTMAXCROSSENTROPYGRAD, "SsdSparseSoftmaxCrossEntropyGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDSQUEEZEFUSION, "SsdSqueezeFusion");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONCATFOUR2FIVE, "ConcatFour2Five");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONCATFIVE2FOUR, "ConcatFive2Four");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDREALDIVTILEMUL, "SSDRealdivTileMul");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SSDSUMMULREALDIVMEAN, "SSDSumMulRealdivMean");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(VARIABLEV2, "VariableV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(VARHANDLEOP, "VarHandleOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TEMPORARYVARIABLE, "TemporaryVariable");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DESTROYTEMPORARYVARIABLE, "DestroyTemporaryVariable");
|
|
|
|
REGISTER_OPTYPE_DEFINE(VARIABLE, "Variable");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASSIGN, "Assign");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASSIGNVARIABLEOP, "AssignVariableOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASSIGNADD, "AssignAdd");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASSIGNADDVARIABLEOP, "AssignAddVariableOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASSIGNSUB, "AssignSub");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASSIGNSUBVARIABLEOP, "AssignSubVariableOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYMOMENTUM, "ApplyMomentum");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RESOURCEAPPLYMOMENTUM, "ResourceApplyMomentum");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SGD, "SGD");
|
|
|
|
REGISTER_OPTYPE_DEFINE(NOOP, "NoOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(READVARIABLEOP, "ReadVariableOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(PARALLELCONCATSTART, "_ParallelConcatStart");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONSTANTOP, "Constant");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEPTHWISECONV2DBACKPROPFILTER, "DepthwiseConv2dNativeBackpropFilter");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEPTHWISECONV2DBACKPORPINPUT, "DepthwiseConv2dNativeBackpropInput");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEPTHWISECONV2DFORWARDNATIVE, "DepthwiseConv2dNative");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DROPOUTGRAD, "DropOutGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYRMSPROPMIXEDPRECISION, "apply_rms_prop_mixed_precision");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYRMSPROP, "ApplyRMSProp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RELU6GRAD, "Relu6Grad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(AVGPOOLGRAD, "AvgPoolGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONCATV2, "ConcatV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONCATOFFSET, "ConcatOffset");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LAYERNORMGRAD, "LayerNormGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LAYERNORM, "LayerNorm");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LARS, "Lars");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DYNAMICSTITCH, "DynamicStitch");
|
|
|
|
|
|
|
|
/***************************************************/
|
|
|
|
REGISTER_OPTYPE_DEFINE(SQUARE, "Square");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMBROADCAST, "HcomBroadcast");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMALLGATHER, "HcomAllGather");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMALLREDUCE, "HcomAllReduce");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMREDUCESCATTER, "HcomReduceScatter");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMREDUCE, "HcomReduce");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMSEND, "HcomSend");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMRECEIVE, "HcomReceive");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMREMOTEREAD, "HcomRemoteRead");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMREMOTEREFREAD, "HcomRemoteRefRead");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMREMOTEWRITE, "HcomRemoteWrite");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HCOMREMOTESCATTERWRITE, "HcomRemoteScatterWrite");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(VARASSIGN, "VarAssign");
|
|
|
|
REGISTER_OPTYPE_DEFINE(VARISINITIALIZEDOP, "VarIsInitializedOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LogTimeStamp, "LogTimeStamp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ISVARIABLEINITIALIZED, "IsVariableInitialized");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STREAMSWITCH, "StreamSwitch");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STREAMSWITCHN, "StreamSwitchN");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STREAMACTIVE, "StreamActive");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MEMCPYASYNC, "MemcpyAsync");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MEMCPYADDRASYNC, "MemcpyAddrAsync");
|
|
|
|
REGISTER_OPTYPE_DEFINE(STREAMMERGE, "StreamMerge");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ENDGRAPH, "EndGraph");
|
|
|
|
REGISTER_OPTYPE_DEFINE(MODELEXIT, "ModelExit");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SEND, "Send");
|
|
|
|
REGISTER_OPTYPE_DEFINE(RECV, "Recv");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ENDOFSEQUENCE, "EndOfSequence");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(LABELSET, "LabelSet");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LABELGOTO, "LabelGoto");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LABELGOTOEX, "LabelGotoEx");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LABELSWITCH, "LabelSwitch");
|
|
|
|
REGISTER_OPTYPE_DEFINE(LABELSWITCHBYINDEX, "LabelSwitchByIndex");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(ATOMICADDRCLEAN, "AtomicAddrClean");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(ABS_GRAD, "AbsGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ACCUMULATE_N_V2, "AccumulateNV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ACOS_GRAD, "AcosGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ACOSH_GRAD, "AcoshGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ANY, "Any");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPROXIMATE_EQUAL, "ApproximateEqual");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASIN_GRAD, "AsinGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ASINH_GRAD, "AsinhGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ATAN_GRAD, "AtanGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BROADCAST_TO, "BroadcastTo");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ELU_GRAD, "EluGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ADD_V2, "AddV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DATAFORMATDIMMAP, "DataFormatDimMap");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DATAFORMATVECPERMUTE, "DataFormatVecPermute");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BESSELI0E, "BesselI0e");
|
|
|
|
REGISTER_OPTYPE_DEFINE(BESSELI1E, "BesselI1e");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYADADELTA, "ApplyAdadelta");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYADAGRAD, "ApplyAdagrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYADAGRADDA, "ApplyAdagradDA");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYADAM, "ApplyAdam");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYADAMAX, "ApplyAdaMax");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYADDSIGN, "ApplyAddSign");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYCENTEREDRMSPROP, "ApplyCenteredRMSProp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYFTRL, "ApplyFtrl");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYFTRLV2, "ApplyFtrlV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYGRADIENTDESCENT, "ApplyGradientDescent");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYPOWERSIGN, "ApplyPowerSign");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYPROXIMALADAGRAD, "ApplyProximalAdagrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(APPLYPROXIMALGRADIENTDESCENT, "ApplyProximalGradientDescent");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEQUANTIZE, "Dequantize");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(FOCAL_LOSS, "FocalLoss");
|
|
|
|
REGISTER_OPTYPE_DEFINE(FOCAL_LOSS_GRAD, "FocalLossGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SMOOTHL1_LOSS, "SmoothL1Loss");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SMOOTHL1_LOSS_grad, "SmoothL1LossGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(REDUCEMEAN, "ReduceMean");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CONCAT_V2, "ConcatV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(ONEHOT_V2, "OneHotV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SLICE_V2, "SliceV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(TILE_V2, "TileV2");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SUM_V2, "SumV2");
|
|
|
|
// Common type when the operator has the same name
|
|
|
|
REGISTER_OPTYPE_DEFINE(DETECTIONOUTPUT, "DetectionOutput");
|
|
|
|
// Custom operator
|
|
|
|
REGISTER_OPTYPE_DEFINE(CUSTOMOP, "CustomOp");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CUSTOMOP_NCHW, "CustomOpNchw");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CUSTOMOP_NHWC, "CustomOpNhwc");
|
|
|
|
REGISTER_OPTYPE_DEFINE(CUSTOMOP_NC1HWC0, "CustomOpNc1hwc0");
|
|
|
|
|
|
|
|
// Depthwise 4d_2_6d,6d_2_4d
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEPTHWISEWEIGHT4D26D, "depthwise_weight_4d_2_6d");
|
|
|
|
REGISTER_OPTYPE_DEFINE(DEPTHWISEWEIGHT6D24D, "depthwise_weight_6d_2_4d");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(SQRTGRAD, "SqrtGrad");
|
|
|
|
REGISTER_OPTYPE_DEFINE(SIGMOIDGRAD, "SigmoidGrad");
|
|
|
|
|
|
|
|
REGISTER_OPTYPE_DEFINE(TRANSSHAPE, "TransShape");
|
|
|
|
|
|
|
|
// Horovod operator
|
|
|
|
REGISTER_OPTYPE_DEFINE(HVDCALLBACKALLREDUCE, "HorovodAllreduce");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HVDCALLBACKALLGATHER, "HorovodAllgather");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HVDCALLBACKBROADCAST, "HorovodBroadcast");
|
|
|
|
REGISTER_OPTYPE_DEFINE(HVDWAIT, "HorovodWait");
|
|
|
|
|
|
|
|
// aicpu op for online_infer dynamic_dims
|
|
|
|
REGISTER_OPTYPE_DEFINE(GETDYNAMICDIMS, "GetDynamicDims");
|
|
|
|
|
|
|
|
// profiling training trace node
|
|
|
|
REGISTER_OPTYPE_DEFINE(PROFILINGTRAININGTRACE, "ProfilingTrainingTrace");
|
|
|
|
|
|
|
|
const std::string MODEL_ATTR_TASKS = "tasks";
|
|
|
|
const std::string MODEL_ATTR_TASK_GEN_BASE_ADDR = "task_gen_base_addr";
|
|
|
|
const std::string MODEL_ATTR_TASK_GEN_WEIGHT_ADDR = "task_gen_weight_addr";
|
|
|
|
const std::string MODEL_ATTR_FUSION_MODEL_DEF = "fm";
|
|
|
|
|
|
|
|
const int MODEL_MAX_SIZE = INT32_MAX; // Max size of 2 GB minus 1 byte.
|
|
|
|
const uint64_t FILE_HEADER_MAX_SIZE = 3221225472; // Max size of 3 GB.
|
|
|
|
|
|
|
|
#if !defined(__ANDROID__) && !defined(ANDROID)
|
|
|
|
const uint64_t ALLOC_MEMORY_MAX_SIZE = 8589934592; // Max size of 8 GB.
|
|
|
|
#else
|
|
|
|
const uint64_t ALLOC_MEMORY_MAX_SIZE = 536870912; // Max size of 512M.
|
|
|
|
#endif
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @brief Magic number of model file
|
|
|
|
///
|
|
|
|
const uint32_t MODEL_FILE_MAGIC_NUM = 0x444F4D49; // magic number
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @brief Model head length
|
|
|
|
///
|
|
|
|
const uint32_t MODEL_FILE_HEAD_LEN = 256;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Input node type
|
|
|
|
///
|
|
|
|
const std::string INPUT_TYPE = "Input";
|
|
|
|
const std::string DUMMY_DATA = "DummyData";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief AIPP label, label AIPP conv operator
|
|
|
|
///
|
|
|
|
const std::string AIPP_CONV_FLAG = "Aipp_Conv_Flag";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief AIPP label, label aipp data operator
|
|
|
|
///
|
|
|
|
const std::string AIPP_DATA_FLAG = "Aipp_Data_Flag";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Record the w dimension of model input corresponding to dynamic AIPP
|
|
|
|
///
|
|
|
|
const std::string AIPP_RELATED_DATA_DIM_W = "aipp_related_data_dim_w";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Record the H dimension of model input corresponding to dynamic AIPP
|
|
|
|
///
|
|
|
|
const std::string AIPP_RELATED_DATA_DIM_H = "aipp_related_data_dim_h";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief The tag of the data operator. Mark this input to the dynamic AIPP operator
|
|
|
|
///
|
|
|
|
const std::string INPUT_TO_DYNAMIC_AIPP = "input_to_dynamic_aipp";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief DATA node type
|
|
|
|
///
|
|
|
|
const std::string DATA_TYPE = "Data";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief DATA node type
|
|
|
|
///
|
|
|
|
const std::string AIPP_DATA_TYPE = "AippData";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Frame operator type
|
|
|
|
///
|
|
|
|
const std::string FRAMEWORK_OP_TYPE = "FrameworkOp";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Data node type
|
|
|
|
///
|
|
|
|
const std::string ANN_DATA_TYPE = "AnnData";
|
|
|
|
const std::string ANN_NETOUTPUT_TYPE = "AnnNetOutput";
|
|
|
|
const std::string ANN_DEPTHCONV_TYPE = "AnnDepthConv";
|
|
|
|
const std::string ANN_CONV_TYPE = "AnnConvolution";
|
|
|
|
const std::string ANN_FC_TYPE = "AnnFullConnection";
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Convolution node type
|
|
|
|
///
|
|
|
|
const std::string NODE_NAME_NET_OUTPUT = "Node_Output";
|
|
|
|
|
|
|
|
const std::string NODE_NAME_END_GRAPH = "Node_EndGraph";
|
|
|
|
|
|
|
|
const std::string NODE_NAME_OP_DEBUG = "Node_OpDebug";
|
|
|
|
const std::string OP_TYPE_OP_DEBUG = "Opdebug";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Convolution node type
|
|
|
|
///
|
|
|
|
const std::string OP_TYPE_CONVOLUTION = "Convolution";
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Add convolution node name to AIPP
|
|
|
|
///
|
|
|
|
const std::string AIPP_CONV_OP_NAME = "aipp_conv_op";
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Operator configuration item separator
|
|
|
|
///
|
|
|
|
const std::string OP_CONF_DELIMITER = ":";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief attr value name
|
|
|
|
///
|
|
|
|
const std::string ATTR_NAME_VALUE1 = "value1";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief attr value name, 6d_2_4d C
|
|
|
|
///
|
|
|
|
const std::string ATTR_NAME_INPUT_CVALUE = "input_cvalue";
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief alpha default value
|
|
|
|
///
|
|
|
|
const float ALPHA_DEFAULT_VALUE = 1.0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief beta default value
|
|
|
|
///
|
|
|
|
const float BETA_DEFAULT_VALUE = 0.0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief coef default value
|
|
|
|
///
|
|
|
|
const float COEF_DEFAULT_VALUE = 0.0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Relu6 coef value
|
|
|
|
///
|
|
|
|
const float RELU6_COEF = 6.0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief stride default value
|
|
|
|
///
|
|
|
|
const uint32_t STRIDE_DEFAULT_VALUE = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief pad default value
|
|
|
|
///
|
|
|
|
const uint32_t PAD_DEFAULT_VALUE = 0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief dilation default value
|
|
|
|
///
|
|
|
|
const int DILATION_DEFAULT_VALUE = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief kernel default value
|
|
|
|
///
|
|
|
|
const uint32_t KERNEL_DEFAULT_VALUE = 0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief defaule convolution group size
|
|
|
|
///
|
|
|
|
const uint32_t DEFAULT_CONV_GROUP = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Default deconvolution adj
|
|
|
|
///
|
|
|
|
const uint32_t DEFAULT_DECONV_ADJ = 0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Represents value 1
|
|
|
|
///
|
|
|
|
const uint32_t NUM_ONE = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief spatial dim size default value
|
|
|
|
///
|
|
|
|
const int32_t SPATIAL_DIM_DEFAULT_SIZE = 2;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief dim extended default value
|
|
|
|
///
|
|
|
|
const int32_t DIM_DEFAULT_VALUE = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief The first weight list in opdef is filter
|
|
|
|
///
|
|
|
|
const int32_t WEIGHT_FILTER_INDEX = 0;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief The second weight list in opdef is bias
|
|
|
|
///
|
|
|
|
const int32_t WEIGHT_BIAS_INDEX = 1;
|
|
|
|
|
|
|
|
const int32_t TENSOR_ND_SUPPORT_SIZE = 8;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief NCHW index default value
|
|
|
|
///
|
|
|
|
const uint32_t NCHW_DIM_N = 0;
|
|
|
|
const uint32_t NCHW_DIM_C = 1;
|
|
|
|
const uint32_t NCHW_DIM_H = 2;
|
|
|
|
const uint32_t NCHW_DIM_W = 3;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief KCHW index default value
|
|
|
|
///
|
|
|
|
const uint32_t KCHW_DIM_K = 0;
|
|
|
|
const uint32_t KCHW_DIM_C = 1;
|
|
|
|
const uint32_t KCHW_DIM_H = 2;
|
|
|
|
const uint32_t KCHW_DIM_W = 3;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief HWCK index default value
|
|
|
|
///
|
|
|
|
const uint32_t HWCK_DIM_H = 0;
|
|
|
|
const uint32_t HWCK_DIM_W = 1;
|
|
|
|
const uint32_t HWCK_DIM_C = 2;
|
|
|
|
const uint32_t HWCK_DIM_K = 3;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief NHWC index default value
|
|
|
|
///
|
|
|
|
const uint32_t NHWC_DIM_N = 0;
|
|
|
|
const uint32_t NHWC_DIM_H = 1;
|
|
|
|
const uint32_t NHWC_DIM_W = 2;
|
|
|
|
const uint32_t NHWC_DIM_C = 3;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief CHWN index default value
|
|
|
|
///
|
|
|
|
const uint32_t CHWN_DIM_N = 3;
|
|
|
|
const uint32_t CHWN_DIM_C = 0;
|
|
|
|
const uint32_t CHWN_DIM_H = 1;
|
|
|
|
const uint32_t CHWN_DIM_W = 2;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief CHW index default value
|
|
|
|
///
|
|
|
|
const uint32_t CHW_DIM_C = 0;
|
|
|
|
const uint32_t CHW_DIM_H = 1;
|
|
|
|
const uint32_t CHW_DIM_W = 2;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief HWC index default value
|
|
|
|
///
|
|
|
|
const uint32_t HWC_DIM_H = 0;
|
|
|
|
const uint32_t HWC_DIM_W = 1;
|
|
|
|
const uint32_t HWC_DIM_C = 2;
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief Pad index default value
|
|
|
|
///
|
|
|
|
const uint32_t PAD_H_HEAD = 0;
|
|
|
|
const uint32_t PAD_H_TAIL = 1;
|
|
|
|
const uint32_t PAD_W_HEAD = 2;
|
|
|
|
const uint32_t PAD_W_TAIL = 3;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief window index default value
|
|
|
|
///
|
|
|
|
const uint32_t WINDOW_H = 0;
|
|
|
|
const uint32_t WINDOW_W = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief stride index default value
|
|
|
|
///
|
|
|
|
const uint32_t STRIDE_H = 0;
|
|
|
|
const uint32_t STRIDE_W = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief dilation index default value
|
|
|
|
///
|
|
|
|
const uint32_t DILATION_H = 0;
|
|
|
|
const uint32_t DILATION_W = 1;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief the num of XRBG channel
|
|
|
|
///
|
|
|
|
const uint32_t XRGB_CHN_NUM = 4;
|
|
|
|
|
|
|
|
///
|
|
|
|
/// @ingroup domi_omg
|
|
|
|
/// @brief global pooling default value
|
|
|
|
///
|
|
|
|
const bool DEFAULT_GLOBAL_POOLING = false;
|
|
|
|
|
|
|
|
const uint32_t MODEL_VERSION = 0x20000000; ///< Model version 2.0///
|
|
|
|
|
|
|
|
// Eltwise's input size
|
|
|
|
const int ELTWISE_MIN_INPUT_SIZE = 2;
|
|
|
|
|
|
|
|
// flowctrl
|
|
|
|
const std::string NODE_NAME_STREAM_SWITCH = "IteratorCtrl_StreamSwitch";
|
|
|
|
const std::string NODE_NAME_STREAM_ACTIVE = "IteratorCtrl_StreamActive";
|
|
|
|
const std::string NODE_NAME_FLOWCTRL_LOOP_PER_ITER = "npu_runconfig/iterations_per_loop";
|
|
|
|
const std::string NODE_NAME_FLOWCTRL_LOOP_COND = "npu_runconfig/loop_cond";
|
|
|
|
const std::string NODE_NAME_FLOWCTRL_LOOP_INCREMENT = "npu_runconfig/one";
|
|
|
|
const std::string NODE_NAME_FLOWCTRL_LOOP_RESETVALUE = "npu_runconfig/zero";
|
|
|
|
const std::string NODE_NAME_FLOWCTRL_LOOP_ASSIGNADD = "FlowCtrl_LoopCond_ASSIGNADD";
|
|
|
|
const std::string NODE_NAME_FLOWCTRL_LOOP_ASSIGN = "FlowCtrl_LoopCond_ASSIGN";
|
|
|
|
const std::string NODE_NAME_ATOMIC_ADDR_CLEAN = "atomic_addr_clean";
|
|
|
|
const uint32_t TRUE_STREAM_ID = 0;
|
|
|
|
const uint32_t STREAM_SWITCH_INPUT_NUM = 2;
|
|
|
|
|
|
|
|
const std::string NODE_NAME_GLOBAL_STEP = "ge_global_step";
|
|
|
|
const std::string NODE_NAME_GLOBAL_STEP_ASSIGNADD = "global_step_assignadd";
|
|
|
|
} // namespace ge
|