Browse Source

!8612 [MS][LITE]control flow schema ops

From: @YeFeng_24
Reviewed-by: @hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
e8c8b40e5a
2 changed files with 44 additions and 1 deletions
  1. +10
    -1
      mindspore/lite/schema/model.fbs
  2. +34
    -0
      mindspore/lite/schema/ops.fbs

+ 10
- 1
mindspore/lite/schema/model.fbs View File

@@ -236,7 +236,16 @@ union PrimitiveType {
LpNormalization, LpNormalization,
DropoutGrad, DropoutGrad,
MaximumGrad, MaximumGrad,
MinimumGrad
MinimumGrad,
Switch,
Partial,
TensorListFromTensor,
TensorListStack,
TensorListGetItem,
TensorListSetItem,
TensorListReserve,
All,
Assert,
} }


enum QuantType: int { enum QuantType: int {


+ 34
- 0
mindspore/lite/schema/ops.fbs View File

@@ -1143,3 +1143,37 @@ table LpNormalization {
axis : int; axis : int;
p : 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;
}

Loading…
Cancel
Save