fix_ops_not_equal

5 years ago · ccdee9549d
parent 9cb1ae8ab9
commit ccdee9549d
1 changed files with 2 additions and 2 deletions
--- 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;
 }