diff --git a/mindspore/lite/src/ops/not_equal.cc b/mindspore/lite/src/ops/not_equal.cc index b768d74293e3d0ac53afc201dc0b8b5ff7a112ee..af97dae95ab7daac377827b6b1f6c7dc0e37e651 100644 --- a/mindspore/lite/src/ops/not_equal.cc +++ b/mindspore/lite/src/ops/not_equal.cc @@ -23,8 +23,8 @@ namespace lite { int NotEqual::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) { MS_ASSERT(nullptr != primitive); MS_ASSERT(nullptr != fbb); - auto val_offset = schema::CreateSin(*fbb); - auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_Sin, val_offset.o); + auto val_offset = schema::CreateNotEqual(*fbb); + auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_NotEqual, val_offset.o); fbb->Finish(prim_offset); return RET_OK; }