From f631889a4768276495c8eaabf64349b69b75f956 Mon Sep 17 00:00:00 2001
From: yefeng <yefeng24@huawei.com>
Date: Mon, 16 Nov 2020 09:30:11 +0800
Subject: [PATCH] control_flow_schema_ops

---
 mindspore/lite/schema/model.fbs | 11 ++++++++++-
 mindspore/lite/schema/ops.fbs   | 34 +++++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/mindspore/lite/schema/model.fbs b/mindspore/lite/schema/model.fbs
index 26c31f51ea..70ab58645a 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 5a768af145..93cb226e8d 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