diff --git a/mindspore/lite/schema/ops.fbs b/mindspore/lite/schema/ops.fbs index ad77ac2526..a7f6a930e7 100644 --- a/mindspore/lite/schema/ops.fbs +++ b/mindspore/lite/schema/ops.fbs @@ -1033,28 +1033,24 @@ table Size { } table RandomStandardNormal { - seed : int; - seed2 : int; + seed: long; + seed2: long; } table CropAndResize { - method : ResizeMethod; - extrapolation_value : float; + method: ResizeMethod; + extrapolation_value: float; } -table StridedSliceGrad { - begin_mask: int; - end_mask: int; - ellipsis_mask: int; - new_axis_mask: int; - shrink_axis_mask: int; - begin: [int]; - end: [int]; - stride: [int]; - isScale: [int]; +table Erf { } -table Erf { +table StridedSliceGrad { + begin_mask: long; + end_mask: long; + ellipsis_mask: long; + new_axis_mask: long; + shrink_axis_mask: long; } table IsFinite { @@ -1064,8 +1060,9 @@ table LinSpace { } table UniformReal { - seed : int; - seed2 : int; + seed: long; + seed2: long; } + table AbsGrad { } diff --git a/mindspore/lite/src/ops/ops_def.cc b/mindspore/lite/src/ops/ops_def.cc index 590bf935e9..b79106711f 100644 --- a/mindspore/lite/src/ops/ops_def.cc +++ b/mindspore/lite/src/ops/ops_def.cc @@ -188,7 +188,7 @@ OP_TYPE(Where) OP_TYPE(ZerosLike) OP_TYPE(Select) OP_TYPE(If) -OP_TYPE(Gru) +OP_TYPE(GRU) OP_TYPE(NonZero) OP_TYPE(InvertPermutation) OP_TYPE(Size)