|
|
@@ -30,74 +30,74 @@ table Blob { |
|
|
|
table Reserved0 {} |
|
|
|
|
|
|
|
union OperatorParam { |
|
|
|
param.Empty, |
|
|
|
param.Axis, |
|
|
|
param.Convolution, |
|
|
|
param.MaskPropagate, |
|
|
|
param.ConvPooling, |
|
|
|
param.ConvBias, |
|
|
|
param.SeparableConv, |
|
|
|
param.Images2Neibs, |
|
|
|
param.Pooling, |
|
|
|
param.LRN, |
|
|
|
param.BN, |
|
|
|
param.ROIPooling, |
|
|
|
param.WarpPerspective, |
|
|
|
param.SpatialTfGridGenerator, |
|
|
|
param.SpatialTfSampler, |
|
|
|
param.MGBAddUpdate, |
|
|
|
param.Elemwise, |
|
|
|
param.ElemwiseMultiType, |
|
|
|
param.PowC, |
|
|
|
param.MatrixMul, |
|
|
|
param.Winograd, |
|
|
|
param.SVD, |
|
|
|
param.Reduce, |
|
|
|
param.Cumsum, |
|
|
|
param.CondTake, |
|
|
|
param.Argsort, |
|
|
|
param.IndexingRemap, |
|
|
|
param.MGBSleep, |
|
|
|
param.Linspace, |
|
|
|
param.LinspaceFull, |
|
|
|
param.Eye, |
|
|
|
param.UniformRNG, |
|
|
|
param.GaussianRNG, |
|
|
|
param.Flip, |
|
|
|
param.Rotate, |
|
|
|
param.ROICopy, |
|
|
|
param.CvtColor, |
|
|
|
param.WarpAffine, |
|
|
|
param.GaussianBlur, |
|
|
|
param.Resize, |
|
|
|
param.Remap, |
|
|
|
param.Convolution3D, |
|
|
|
param.Conv3DBias, |
|
|
|
param.SeparableConv3D, |
|
|
|
param.TopK, |
|
|
|
param.RelayoutFormat, |
|
|
|
param.SeparableFilter, |
|
|
|
param.LocalShare, |
|
|
|
param.ROIAlign, |
|
|
|
param.DeformablePSROIPooling, |
|
|
|
param.BatchConvBias, |
|
|
|
param.DType, |
|
|
|
param.PersistentOutputStorage, |
|
|
|
param.OptionalAxis, |
|
|
|
param.OptionalAxisV1, |
|
|
|
param.ExecutionPolicy, |
|
|
|
param.AssertEqual, |
|
|
|
Reserved0, |
|
|
|
param.CollectiveComm, |
|
|
|
param.CondExecPred, |
|
|
|
param.CondExecPredLogical, |
|
|
|
param.CondExecMark, |
|
|
|
param.CondExecMerge, |
|
|
|
param.Host2DeviceCopy, |
|
|
|
param.Dimshuffle, |
|
|
|
param.AxisAddRemove, |
|
|
|
param.IndexDescMaskDump, |
|
|
|
DType, |
|
|
|
param.Empty = 1, |
|
|
|
param.Axis = 2, |
|
|
|
param.Convolution = 3, |
|
|
|
param.MaskPropagate = 4, |
|
|
|
param.ConvPooling = 5, |
|
|
|
param.ConvBias = 6, |
|
|
|
param.SeparableConv = 7, |
|
|
|
param.Images2Neibs = 8, |
|
|
|
param.Pooling = 9, |
|
|
|
param.LRN = 10, |
|
|
|
param.BN = 11, |
|
|
|
param.ROIPooling = 12, |
|
|
|
param.WarpPerspective = 13, |
|
|
|
param.SpatialTfGridGenerator = 14, |
|
|
|
param.SpatialTfSampler = 15, |
|
|
|
param.MGBAddUpdate = 16, |
|
|
|
param.Elemwise = 17, |
|
|
|
param.ElemwiseMultiType = 18, |
|
|
|
param.PowC = 19, |
|
|
|
param.MatrixMul = 20, |
|
|
|
param.Winograd = 21, |
|
|
|
param.SVD = 22, |
|
|
|
param.Reduce = 23, |
|
|
|
param.Cumsum = 24, |
|
|
|
param.CondTake = 25, |
|
|
|
param.Argsort = 26, |
|
|
|
param.IndexingRemap = 27, |
|
|
|
param.MGBSleep = 28, |
|
|
|
param.Linspace = 29, |
|
|
|
param.LinspaceFull = 30, |
|
|
|
param.Eye = 31, |
|
|
|
param.UniformRNG = 32, |
|
|
|
param.GaussianRNG = 33, |
|
|
|
param.Flip = 34, |
|
|
|
param.Rotate = 35, |
|
|
|
param.ROICopy = 36, |
|
|
|
param.CvtColor = 37, |
|
|
|
param.WarpAffine = 38, |
|
|
|
param.GaussianBlur = 39, |
|
|
|
param.Resize = 40, |
|
|
|
param.Convolution3D = 41, |
|
|
|
param.Conv3DBias = 42, |
|
|
|
param.SeparableConv3D = 43, |
|
|
|
param.TopK = 44, |
|
|
|
param.RelayoutFormat = 45, |
|
|
|
param.SeparableFilter = 46, |
|
|
|
param.LocalShare = 47, |
|
|
|
param.ROIAlign = 48, |
|
|
|
param.DeformablePSROIPooling = 49, |
|
|
|
param.BatchConvBias = 50, |
|
|
|
param.DType = 51, |
|
|
|
param.PersistentOutputStorage = 52, |
|
|
|
param.OptionalAxis = 53, |
|
|
|
param.OptionalAxisV1 = 54, |
|
|
|
param.ExecutionPolicy = 55, |
|
|
|
param.AssertEqual = 56, |
|
|
|
Reserved0 = 57, |
|
|
|
param.CollectiveComm = 58, |
|
|
|
param.CondExecPred = 59, |
|
|
|
param.CondExecPredLogical = 60, |
|
|
|
param.CondExecMark = 61, |
|
|
|
param.CondExecMerge = 62, |
|
|
|
param.Host2DeviceCopy = 63, |
|
|
|
param.Dimshuffle = 64, |
|
|
|
param.AxisAddRemove = 65, |
|
|
|
param.IndexDescMaskDump = 66, |
|
|
|
DType = 67, |
|
|
|
param.Remap = 68, |
|
|
|
} |
|
|
|
|
|
|
|
table Operator { |
|
|
|