diff --git a/mindspore/lite/schema/model.fbs b/mindspore/lite/schema/model.fbs index 26c31f51eae..70ab58645a7 100644 --- a/mindspore/lite/schema/model.fbs +++ b/mindspore/lite/schema/model.fbs @@ -236,7 +236,16 @@ union PrimitiveType { LpNormalization, DropoutGrad, MaximumGrad, - MinimumGrad + MinimumGrad, + Switch, + Partial, + TensorListFromTensor, + TensorListStack, + TensorListGetItem, + TensorListSetItem, + TensorListReserve, + All, + Assert, } enum QuantType: int { diff --git a/mindspore/lite/schema/ops.fbs b/mindspore/lite/schema/ops.fbs index 5a768af145e..93cb226e8de 100644 --- a/mindspore/lite/schema/ops.fbs +++ b/mindspore/lite/schema/ops.fbs @@ -1143,3 +1143,37 @@ table LpNormalization { axis : int; p : int; } + +table Switch { +} + +table Partial { + subGraphIndex : int; +} + +table TensorListFromTensor { +} + +table TensorListStack { + numElements : int; + elementDType : int; +} + +table TensorListGetItem { + elementDType : int; +} + +table TensorListSetItem { +} + +table TensorListReserve { + elementDType : int; +} + +table All { + keepDims : int; +} + +table Assert { + summarize : int; +} \ No newline at end of file