forked from mindspore-Ecosystem/mindspore
!8612 [MS][LITE]control flow schema ops
From: @YeFeng_24 Reviewed-by: @hangangqiang,@zhanghaibo5 Signed-off-by: @hangangqiang
This commit is contained in:
commit
e8c8b40e5a
|
@ -236,7 +236,16 @@ union PrimitiveType {
|
|||
LpNormalization,
|
||||
DropoutGrad,
|
||||
MaximumGrad,
|
||||
MinimumGrad
|
||||
MinimumGrad,
|
||||
Switch,
|
||||
Partial,
|
||||
TensorListFromTensor,
|
||||
TensorListStack,
|
||||
TensorListGetItem,
|
||||
TensorListSetItem,
|
||||
TensorListReserve,
|
||||
All,
|
||||
Assert,
|
||||
}
|
||||
|
||||
enum QuantType: int {
|
||||
|
|
|
@ -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;
|
||||
}
|
Loading…
Reference in New Issue